remove ver in model gen
Browse files
utils.py
CHANGED
@@ -7,8 +7,8 @@ import torch
|
|
7 |
|
8 |
from huggan.pytorch.lightweight_gan.lightweight_gan import LightweightGAN
|
9 |
|
10 |
-
def load_model(model_name="ceyda/butterfly_cropped_uniq1K_512"
|
11 |
-
gan = LightweightGAN.from_pretrained(model_name
|
12 |
gan.eval()
|
13 |
return gan
|
14 |
|
|
|
7 |
|
8 |
from huggan.pytorch.lightweight_gan.lightweight_gan import LightweightGAN
|
9 |
|
10 |
+
def load_model(model_name="ceyda/butterfly_cropped_uniq1K_512"):
|
11 |
+
gan = LightweightGAN.from_pretrained(model_name)
|
12 |
gan.eval()
|
13 |
return gan
|
14 |
|