Vintern-v2-Demo

Running on Zero

qnguyen3 commited on Jul 4

Commit

e6eaffd

•

1 Parent(s): d5f6db7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ tokenizer = AutoTokenizer.from_pretrained(
     trust_remote_code=True)
 model = LlavaQwen2ForCausalLM.from_pretrained(
-    'qnguyen3/nanoLLaVA',
     torch_dtype=torch.float16,
     attn_implementation="flash_attention_2",
     trust_remote_code=True)

     trust_remote_code=True)
 model = LlavaQwen2ForCausalLM.from_pretrained(
+    'qnguyen3/nanoLLaVA-1.5',
     torch_dtype=torch.float16,
     attn_implementation="flash_attention_2",
     trust_remote_code=True)