vishesh-t27 commited on
Commit
ebbe8c3
·
verified ·
1 Parent(s): b0185c2

updated readme.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -75,7 +75,7 @@ model = AutoModelForCausalLM.from_pretrained(
75
  model_name,
76
  trust_remote_code=True,
77
  dtype=torch.bfloat16
78
- ).device()
79
 
80
  prompt = "Explain newton's second law of motion"
81
 
 
75
  model_name,
76
  trust_remote_code=True,
77
  dtype=torch.bfloat16
78
+ ).to(device).eval()
79
 
80
  prompt = "Explain newton's second law of motion"
81