Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -21,7 +21,7 @@ HEADER = ("""
|
|
21 |
|
22 |
device = "cuda"
|
23 |
model = AutoModelForCausalLM.from_pretrained(
|
24 |
-
"
|
25 |
trust_remote_code=True,
|
26 |
torch_dtype=torch.bfloat16,
|
27 |
attn_implementation="flash_attention_2",
|
|
|
21 |
|
22 |
device = "cuda"
|
23 |
model = AutoModelForCausalLM.from_pretrained(
|
24 |
+
"OpenGVLab/InternVideo2_5_Chat_8B",
|
25 |
trust_remote_code=True,
|
26 |
torch_dtype=torch.bfloat16,
|
27 |
attn_implementation="flash_attention_2",
|