abetlen commited on
Commit
6d7cb3e
1 Parent(s): cac9a19
Files changed (2) hide show
  1. app.py +2 -2
  2. models.py +0 -1
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, revision) in models.MODELS.items():
248
- gradio_helpers.register_download(name, repo, filename, revision)
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
  },