jseb4s commited on
Commit
c91ed2b
·
verified ·
1 Parent(s): bdf0eb0

se agrega el argumento use_auth_token=None

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -6,7 +6,7 @@ from huggan.pytorch.lightweight_gan.lightweight_gan import LightweightGAN
6
 
7
  # Cargar modelo
8
  def carga_modelo(model_name = "ceyda/butterfly_cropped_uniq1K_512", model_version = None):
9
- gan = LightweightGAN.from_pretrained(model_name, version = model_version)
10
  gan.eval()
11
  return gan
12
 
 
6
 
7
  # Cargar modelo
8
  def carga_modelo(model_name = "ceyda/butterfly_cropped_uniq1K_512", model_version = None):
9
+ gan = LightweightGAN.from_pretrained(model_name, version = model_version, use_auth_token=None)
10
  gan.eval()
11
  return gan
12