FreedomIntelligence
/

ALLaVA-3B

Text Generation

Model card Files Files and versions

g-h-chen commited on Feb 29, 2024

Commit

1b2b1ff

·

verified ·

1 Parent(s): 4003a9b

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -15,7 +15,7 @@ from transformers import AutoTokenizer
 import torch
 import pdb
-dir = "FreedomIntelligence/ALLaVA-3B-Longer"
 device = 'cuda'
 model = AutoModelForCausalLM.from_pretrained(dir, trust_remote_code=True, device_map=device, torch_dtype=torch.bfloat16)

 import torch
 import pdb
+dir = "FreedomIntelligence/ALLaVA-3B"
 device = 'cuda'
 model = AutoModelForCausalLM.from_pretrained(dir, trust_remote_code=True, device_map=device, torch_dtype=torch.bfloat16)