feihu.hf
commited on
Commit
·
c08ee2d
1
Parent(s):
e1da15d
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-14B-Chat-AWQ",
|
|
|
|
| 51 |
device_map="auto"
|
| 52 |
)
|
| 53 |
tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen1.5-14B-Chat-AWQ")
|
|
|
|
| 48 |
|
| 49 |
model = AutoModelForCausalLM.from_pretrained(
|
| 50 |
"Qwen/Qwen1.5-14B-Chat-AWQ",
|
| 51 |
+
torch_dtype="auto",
|
| 52 |
device_map="auto"
|
| 53 |
)
|
| 54 |
tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen1.5-14B-Chat-AWQ")
|