Weights for vgg16 are downloaded every time

dribnet / pixray

neural image generation

Other

402 stars 53 forks source link

Closed samedii closed 3 years ago

samedii commented 3 years ago

Command:

cog run python pixray.py \
    --prompt="underwayer city" \
    --drawer=vqgan \
    --quality=draft \
    --aspect=square \

Issue:

Downloading: "https://download.pytorch.org/models/vgg16-397923af.pth" to /root/.cache/torch/hub/checkpoints/vgg16-397923af.pth

samedii commented 3 years ago

This is easily solved by adding

torch.hub.set_dir("models")