leovale14 commited on
Commit
fc64553
·
1 Parent(s): 2b713f3

fix carga_modelo() takes from 0 to 1 positional arguments but 2 were given

Browse files
Files changed (1) hide show
  1. utils.py +2 -2
utils.py CHANGED
@@ -15,7 +15,7 @@ local_files_only = False
15
  token = None
16
 
17
 
18
- def carga_modelo(model_name="ceyda/butterfly_cropped_uniq1K_512"):
19
  """
20
  Loads a pre-trained LightweightGAN model from Hugging Face Model Hub.
21
 
@@ -30,7 +30,7 @@ def carga_modelo(model_name="ceyda/butterfly_cropped_uniq1K_512"):
30
  config_file = hf_hub_download(
31
  repo_id=str(model_name),
32
  filename=CONFIG_NAME,
33
- revision=revision,
34
  cache_dir=cache_dir,
35
  force_download=force_download,
36
  proxies=proxies,
 
15
  token = None
16
 
17
 
18
+ def carga_modelo(model_name="ceyda/butterfly_cropped_uniq1K_512", version_modelo=None):
19
  """
20
  Loads a pre-trained LightweightGAN model from Hugging Face Model Hub.
21
 
 
30
  config_file = hf_hub_download(
31
  repo_id=str(model_name),
32
  filename=CONFIG_NAME,
33
+ revision=version_modelo,
34
  cache_dir=cache_dir,
35
  force_download=force_download,
36
  proxies=proxies,