Spaces:
Sleeping
Sleeping
Commit ·
fc64553
1
Parent(s): 2b713f3
fix carga_modelo() takes from 0 to 1 positional arguments but 2 were given
Browse files
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=
|
| 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,
|