Spaces:
Running
on
Zero
Running
on
Zero
Update demo.py
Browse files
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="
|
22 |
)
|
23 |
|
24 |
|
|
|
18 |
trust_remote_code=True,
|
19 |
torch_dtype=torch.float16,
|
20 |
device_map={"": "cuda"},
|
21 |
+
revision="84db61d0ff2dca21475a11a3c0e3080e2d2723d5"
|
22 |
)
|
23 |
|
24 |
|