Spaces:
Runtime error
Runtime error
Update chatbot.py
Browse files- chatbot.py +1 -1
chatbot.py
CHANGED
|
@@ -30,7 +30,7 @@ from groq import Groq
|
|
| 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")
|
| 35 |
# Credit to merve for code of llava interleave qwen
|
| 36 |
|
|
|
|
| 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")
|
| 35 |
# Credit to merve for code of llava interleave qwen
|
| 36 |
|