feihu.hf commited on
Commit
5dad842
1 Parent(s): f8b29f6

add torch_dtype

Browse files
Files changed (1) hide show
  1. README.md +1 -0
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-72B-Chat-AWQ",
 
51
  device_map="auto"
52
  )
53
  tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen1.5-72B-Chat-AWQ")
 
48
 
49
  model = AutoModelForCausalLM.from_pretrained(
50
  "Qwen/Qwen1.5-72B-Chat-AWQ",
51
+ torch_dtype="auto",
52
  device_map="auto"
53
  )
54
  tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen1.5-72B-Chat-AWQ")