lixin4ever commited on
Commit
674be41
·
verified ·
1 Parent(s): 593a1aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ model = AutoModelForCausalLM.from_pretrained(
19
  attn_implementation="flash_attention_2",
20
  )
21
  model.to(device)
22
- processor = AutoProcessor.from_pretrained(model_name, trust_remote_code=True)
23
 
24
 
25
  class VideoLLaMA3GradioInterface(object):
 
19
  attn_implementation="flash_attention_2",
20
  )
21
  model.to(device)
22
+ processor = AutoProcessor.from_pretrained("DAMO-NLP-SG/VideoLLaMA3-7B", trust_remote_code=True)
23
 
24
 
25
  class VideoLLaMA3GradioInterface(object):