abetlen commited on
Commit
cac9a19
·
1 Parent(s): b206e22
Files changed (1) hide show
  1. models.py +2 -3
models.py CHANGED
@@ -12,14 +12,13 @@ import gradio_helpers
12
 
13
  ORGANIZATION = 'abetlen'
14
  BASE_MODELS = [
15
- ('paligemma-3b-mix-224-jax', 'paligemma-3b-mix-224'),
16
- ('paligemma-3b-mix-448-jax', 'paligemma-3b-mix-448'),
17
  ]
18
  MODELS = {
19
  **{
20
  model_name: (
21
  f'{ORGANIZATION}/{repo}',
22
- f'{model_name}.bf16.npz',
23
  'bfloat16', # Model repo revision.
24
  )
25
  for repo, model_name in BASE_MODELS
 
12
 
13
  ORGANIZATION = 'abetlen'
14
  BASE_MODELS = [
15
+ ('paligemma-3b-mix-224-gguf', 'paligemma-3b-mix-224-text-model-q4_k_m.gguf'),
 
16
  ]
17
  MODELS = {
18
  **{
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