Spaces:
Running
Running
Update
Browse files
app.py
CHANGED
|
@@ -244,7 +244,7 @@ if __name__ == '__main__':
|
|
| 244 |
logging.info('environ["%s"] = %r', k, v)
|
| 245 |
|
| 246 |
gradio_helpers.set_warmup_function(warmup)
|
| 247 |
-
for name, (repo, filename
|
| 248 |
-
gradio_helpers.register_download(name, repo, filename
|
| 249 |
|
| 250 |
create_app().queue().launch()
|
|
|
|
| 244 |
logging.info('environ["%s"] = %r', k, v)
|
| 245 |
|
| 246 |
gradio_helpers.set_warmup_function(warmup)
|
| 247 |
+
for name, (repo, filename) in models.MODELS.items():
|
| 248 |
+
gradio_helpers.register_download(name, repo, filename)
|
| 249 |
|
| 250 |
create_app().queue().launch()
|
models.py
CHANGED
|
@@ -19,7 +19,6 @@ MODELS = {
|
|
| 19 |
model_name: (
|
| 20 |
f'{ORGANIZATION}/{repo}',
|
| 21 |
f'{model_name}.gguf',
|
| 22 |
-
'bfloat16', # Model repo revision.
|
| 23 |
)
|
| 24 |
for repo, model_name in BASE_MODELS
|
| 25 |
},
|
|
|
|
| 19 |
model_name: (
|
| 20 |
f'{ORGANIZATION}/{repo}',
|
| 21 |
f'{model_name}.gguf',
|
|
|
|
| 22 |
)
|
| 23 |
for repo, model_name in BASE_MODELS
|
| 24 |
},
|