Update
Browse files
models.py
CHANGED
@@ -12,14 +12,13 @@ import gradio_helpers
|
|
12 |
|
13 |
ORGANIZATION = 'abetlen'
|
14 |
BASE_MODELS = [
|
15 |
-
('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}.
|
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
|