qnguyen3 commited on
Commit
e11a252
·
verified ·
1 Parent(s): 10fe8c6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -21,12 +21,12 @@ torch.set_default_device('cuda') # or 'cuda'
21
 
22
  # create model
23
  model = AutoModelForCausalLM.from_pretrained(
24
- 'qnguyen3/siglip-quyen-0.5b-chatml',
25
  torch_dtype=torch.float16,
26
  device_map='auto',
27
  trust_remote_code=True)
28
  tokenizer = AutoTokenizer.from_pretrained(
29
- 'qnguyen3/siglip-quyen-0.5b-chatml',
30
  trust_remote_code=True)
31
 
32
  # text prompt
 
21
 
22
  # create model
23
  model = AutoModelForCausalLM.from_pretrained(
24
+ 'qnguyen3/nanoLLaVA',
25
  torch_dtype=torch.float16,
26
  device_map='auto',
27
  trust_remote_code=True)
28
  tokenizer = AutoTokenizer.from_pretrained(
29
+ 'qnguyen3/nanoLLaVA',
30
  trust_remote_code=True)
31
 
32
  # text prompt