danielkim commited on
Commit
994b152
1 Parent(s): 407810f

Model card has typo

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -54,7 +54,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=False)
54
 
55
  model = AutoModelForCausalLM.from_pretrained(
56
  model_name,
57
- device_map="auto"
58
  torch_dtype=torch.float16,
59
  )
60
 
 
54
 
55
  model = AutoModelForCausalLM.from_pretrained(
56
  model_name,
57
+ device_map="auto",
58
  torch_dtype=torch.float16,
59
  )
60