AdamyaG commited on
Commit
4545b90
·
verified ·
1 Parent(s): b589650

Update chatbot.py

Browse files
Files changed (1) hide show
  1. 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
- # 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")
 
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")