wgcv commited on
Commit
d7fed5a
1 Parent(s): 733e8c1

remove ver in model gen

Browse files
Files changed (1) hide show
  1. utils.py +2 -2
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", model_version=None):
11
- gan = LightweightGAN.from_pretrained(model_name, version=model_version)
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