Update README.md
Browse files
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/
|
25 |
torch_dtype=torch.float16,
|
26 |
device_map='auto',
|
27 |
trust_remote_code=True)
|
28 |
tokenizer = AutoTokenizer.from_pretrained(
|
29 |
-
'qnguyen3/
|
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
|