vikhyatk commited on
Commit
c5f0574
·
verified ·
1 Parent(s): fef07ac

Update demo.py

Browse files
Files changed (1) hide show
  1. demo.py +1 -1
demo.py CHANGED
@@ -18,7 +18,7 @@ model = AutoModelForCausalLM.from_pretrained(
18
  trust_remote_code=True,
19
  torch_dtype=torch.float16,
20
  device_map={"": "cuda"},
21
- revision="69420e0c6596863b4f0059e365fadc5cb388e8fd"
22
  )
23
 
24
 
 
18
  trust_remote_code=True,
19
  torch_dtype=torch.float16,
20
  device_map={"": "cuda"},
21
+ revision="84db61d0ff2dca21475a11a3c0e3080e2d2723d5"
22
  )
23
 
24