Spaces:
Runtime error
Runtime error
Update chatbot.py
Browse files- chatbot.py +2 -2
chatbot.py
CHANGED
@@ -26,9 +26,9 @@ from gradio_client import Client, file
|
|
26 |
from groq import Groq
|
27 |
|
28 |
# You can also use models that are commented below
|
29 |
-
model_id = "llava-hf/llava-interleave-qwen-0.5b-hf"
|
30 |
# model_id = "llava-hf/llava-interleave-qwen-7b-hf"
|
31 |
-
|
32 |
processor = LlavaProcessor.from_pretrained(model_id)
|
33 |
model = LlavaForConditionalGeneration.from_pretrained(model_id,torch_dtype=torch.float16) #, use_flash_attention_2=True)
|
34 |
model.to("cpu")
|
|
|
26 |
from groq import Groq
|
27 |
|
28 |
# You can also use models that are commented below
|
29 |
+
# model_id = "llava-hf/llava-interleave-qwen-0.5b-hf"
|
30 |
# model_id = "llava-hf/llava-interleave-qwen-7b-hf"
|
31 |
+
model_id = "llava-hf/llava-interleave-qwen-7b-dpo-hf"
|
32 |
processor = LlavaProcessor.from_pretrained(model_id)
|
33 |
model = LlavaForConditionalGeneration.from_pretrained(model_id,torch_dtype=torch.float16) #, use_flash_attention_2=True)
|
34 |
model.to("cpu")
|