Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_dir)
|
|
20 |
model = AutoModelForCausalLM.from_pretrained(
|
21 |
model_dir,
|
22 |
torch_dtype=torch.float32,
|
23 |
-
|
24 |
)
|
25 |
|
26 |
# 入力を調整する関数
|
|
|
20 |
model = AutoModelForCausalLM.from_pretrained(
|
21 |
model_dir,
|
22 |
torch_dtype=torch.float32,
|
23 |
+
use_safetensors=True # SafeTensorsを有効化
|
24 |
)
|
25 |
|
26 |
# 入力を調整する関数
|