Spaces:

UniverseTBD
/

astroLLaVA

Sleeping

Sharaf Zaman commited on Sep 30

Commit

155c5ed

•

1 Parent(s): dda2b14

Try llava-hf again.

Files changed (1) hide show

app.py CHANGED Viewed

@@ -48,7 +48,7 @@ def start_worker(model_path: str, bits=16):
         "--model-name",
         model_name,
         "--model-base",
-        "liuhaotian/llava-1.5-7b",
         "--use-flash-attn",
     ]
     # if bits != 16:

         "--model-name",
         model_name,
         "--model-base",
+        "llava-hf/llava-1.5-7b-hf",
         "--use-flash-attn",
     ]
     # if bits != 16: