Vintern-3B-Demo

Running on Zero

qnguyen3 commited on Jul 4

Commit

4d18fd2

•

1 Parent(s): 917a798

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENT
 torch.set_default_device('cuda')
 tokenizer = AutoTokenizer.from_pretrained(
-    'qnguyen3/nanoLLaVA',
     trust_remote_code=True)
 model = LlavaQwen2ForCausalLM.from_pretrained(

 torch.set_default_device('cuda')
 tokenizer = AutoTokenizer.from_pretrained(
+    'qnguyen3/nanoLLaVA-1.5',
     trust_remote_code=True)
 model = LlavaQwen2ForCausalLM.from_pretrained(