Spaces:

khang119966
/

Vintern-v2-Demo

Runtime error

qnguyen3 commited on Jul 4, 2024

Commit

4d18fd2

verified ·

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(