Spaces:

AIDC-AI
/

Ovis2-8B

Running on Zero

xxyyy123 commited on Feb 17

Commit

40ebaf1

verified ·

1 Parent(s): 6a5d716

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ import gradio as gr
 from transformers import AutoModelForCausalLM, TextIteratorStreamer
 model_name = 'AIDC-AI/Ovis2-8B'
-use_thread = True
 # load model
 model = AutoModelForCausalLM.from_pretrained(model_name,

 from transformers import AutoModelForCausalLM, TextIteratorStreamer
 model_name = 'AIDC-AI/Ovis2-8B'
+use_thread = False
 # load model
 model = AutoModelForCausalLM.from_pretrained(model_name,