Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -51,12 +51,6 @@ for MODEL_NAME in MODEL_NAMES:
|
|
51 |
)
|
52 |
MODELS[MODEL_NAME] = synthesizer
|
53 |
|
54 |
-
# Generators
|
55 |
-
#generator2 = gr.Interface.load("huggingface/EleutherAI/gpt-neo-2.7B")
|
56 |
-
#generator3 = gr.Interface.load("huggingface/EleutherAI/gpt-j-6B")
|
57 |
-
#generator1 = gr.Interface.load("huggingface/gpt2-large")
|
58 |
-
|
59 |
-
|
60 |
GEN_NAMES = [
|
61 |
"huggingface/EleutherAI/gpt-neo-2.7B",
|
62 |
"huggingface/EleutherAI/gpt-j-6B",
|
@@ -64,7 +58,7 @@ GEN_NAMES = [
|
|
64 |
]
|
65 |
GENERATORS = {}
|
66 |
manager = ModelManager()
|
67 |
-
for MODEL_NAME in
|
68 |
print(f"downloading {MODEL_NAME}")
|
69 |
model_path, config_path, model_item = manager.download_model(f"{MODEL_NAME}")
|
70 |
#vocoder_name: Optional[str] = model_item["default_vocoder"]
|
|
|
51 |
)
|
52 |
MODELS[MODEL_NAME] = synthesizer
|
53 |
|
|
|
|
|
|
|
|
|
|
|
|
|
54 |
GEN_NAMES = [
|
55 |
"huggingface/EleutherAI/gpt-neo-2.7B",
|
56 |
"huggingface/EleutherAI/gpt-j-6B",
|
|
|
58 |
]
|
59 |
GENERATORS = {}
|
60 |
manager = ModelManager()
|
61 |
+
for MODEL_NAME in GEN_NAMES :
|
62 |
print(f"downloading {MODEL_NAME}")
|
63 |
model_path, config_path, model_item = manager.download_model(f"{MODEL_NAME}")
|
64 |
#vocoder_name: Optional[str] = model_item["default_vocoder"]
|