LinkangZhan commited on
Commit
7c8d14e
1 Parent(s): f595202
Files changed (1) hide show
  1. app.py +1 -1
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.float32,
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),