Spaces:

wgcv
/

img-gen

Sleeping

wgcv commited on Jun 24, 2024

Commit

714d9d4

verified ·

1 Parent(s): d7fed5a

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -8,7 +8,7 @@ import torch
 from huggan.pytorch.lightweight_gan.lightweight_gan import LightweightGAN
 def load_model(model_name="ceyda/butterfly_cropped_uniq1K_512"):
-    gan = LightweightGAN.from_pretrained(model_name)
     gan.eval()
     return gan

 from huggan.pytorch.lightweight_gan.lightweight_gan import LightweightGAN
 def load_model(model_name="ceyda/butterfly_cropped_uniq1K_512"):
+    gan = LightweightGAN.from_pretrained(model_name, use_auth_token=False)
     gan.eval()
     return gan