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 |
},
|