Missing Torch

[RikNieuwoudt]

I tried running this cog on Lambda but for the following on trying to download the weights

Traceback (most recent call last):
  File "script/download-weights", line 6, in <module>
    import torch
ModuleNotFoundError: No module named 'torch'

I pulled the repo, installed Cog and followed the readme.