Spaces:

lixin4ever
/

VideoLLaMA2

Running on Zero

ClownRat commited on Aug 8

Commit

5f06f81

•

1 Parent(s): 6c205ce

Fix bugs.

Files changed (1) hide show

app.py CHANGED Viewed

@@ -157,7 +157,7 @@ model_path = 'DAMO-NLP-SG/VideoLLaMA2-7B-16F'
 device = torch.device("cuda")
-handler = Chat(model_path, conv_mode=conv_mode, load_8bit=False, load_4bit=True)
 textbox = gr.Textbox(show_label=False, placeholder="Enter text and press ENTER", container=False)

 device = torch.device("cuda")
+handler = Chat(model_path, load_8bit=False, load_4bit=True)
 textbox = gr.Textbox(show_label=False, placeholder="Enter text and press ENTER", container=False)