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),
|