Spaces:
Running
Running
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,
|