Spaces:
Runtime error
Runtime error
LinkangZhan
commited on
Commit
·
7c8d14e
1
Parent(s):
f595202
fix gpu
Browse files
app.py
CHANGED
|
@@ -21,7 +21,7 @@ import torch
|
|
| 21 |
# ("Junity/Genshin-World-Model" if lora_folder == ''
|
| 22 |
# else lora_folder),
|
| 23 |
# device_map="auto",
|
| 24 |
-
# torch_dtype=torch.
|
| 25 |
# trust_remote_code=True)
|
| 26 |
# tokenizer = AutoTokenizer.from_pretrained(("baichuan-inc/Baichuan-13B-Base" if model_folder == ''
|
| 27 |
# else model_folder),
|
|
|
|
| 21 |
# ("Junity/Genshin-World-Model" if lora_folder == ''
|
| 22 |
# else lora_folder),
|
| 23 |
# device_map="auto",
|
| 24 |
+
# torch_dtype=torch.float16,
|
| 25 |
# trust_remote_code=True)
|
| 26 |
# tokenizer = AutoTokenizer.from_pretrained(("baichuan-inc/Baichuan-13B-Base" if model_folder == ''
|
| 27 |
# else model_folder),
|