Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -50,7 +50,7 @@ def convert_and_deploy(url, repo_id, hf_token):
|
|
50 |
|
51 |
# モデルをfloat16形式で保存
|
52 |
try:
|
53 |
-
model.save_pretrained("
|
54 |
except Exception as e:
|
55 |
return f"モデルの保存に失敗しました。エラー: {e}"
|
56 |
|
|
|
50 |
|
51 |
# モデルをfloat16形式で保存
|
52 |
try:
|
53 |
+
model.save_pretrained(f"{model_name}_float16", torch_dtype=torch.float16)
|
54 |
except Exception as e:
|
55 |
return f"モデルの保存に失敗しました。エラー: {e}"
|
56 |
|