feihu.hf
commited on
Commit
•
f055cb5
1
Parent(s):
15bf46b
add torch_dtype
Browse files
README.md
CHANGED
|
@@ -48,6 +48,7 @@ device = "cuda" # the device to load the model onto
|
|
| 48 |
|
| 49 |
model = AutoModelForCausalLM.from_pretrained(
|
| 50 |
"Qwen/Qwen1.5-4B-Chat",
|
|
|
|
| 51 |
device_map="auto"
|
| 52 |
)
|
| 53 |
tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen1.5-4B-Chat")
|
|
|
|
| 48 |
|
| 49 |
model = AutoModelForCausalLM.from_pretrained(
|
| 50 |
"Qwen/Qwen1.5-4B-Chat",
|
| 51 |
+
torch_dtype="auto",
|
| 52 |
device_map="auto"
|
| 53 |
)
|
| 54 |
tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen1.5-4B-Chat")
|