MULT

app.py

@@ -37,6 +37,9 @@ COMMENT = {
     "U64": "INT64の処理能力が高いGPUを選定してください。",
     "F64": "FP64の処理能力が高いGPUを選定してください。",
 }
+INF_MULT = 1.25
+LORA_MULT = 1.6
+FULL_MULT = 4
 
 def get_model_api_info(model_name: str) -> dict:
     """
@@ -96,9 +99,9 @@ def estimate_gpu_memory(model_name: str) -> str:
             result_lines.append(f"参考までに、該当リポジトリのファイルサイズは合計 {usedStorage_gb:.2f} GBです。")
             result_lines.append("")
             result_lines.append("これを全てモデルのデータとして仮定した場合、推論及び、トレーニングに必要な概算のGPUメモリサイズは")
-            result_lines.append(f"【推論】約 {usedStorage_gb * 1.3:.2f} GB")
-            result_lines.append(f"【トレーニング（LoRA）】約 {usedStorage_gb * 1.8:.2f} GB")
-            result_lines.append(f"【トレーニング（フルパラメータ）】約 {usedStorage_gb * 4:.2f} GB")
+            result_lines.append(f"【推論】約 {usedStorage_gb * INF_MULT:.2f} GB")
+            result_lines.append(f"【トレーニング（LoRA）】約 {usedStorage_gb * LORA_MULT:.2f} GB")
+            result_lines.append(f"【トレーニング（フルパラメータ）】約 {usedStorage_gb * FULL_MULT:.2f} GB")
             result_lines.append("となります。")
         return "\n".join(result_lines)
 
@@ -123,9 +126,9 @@ def estimate_gpu_memory(model_name: str) -> str:
     result_lines.append(f"これらを合算するとモデルのデータサイズは約 {estimated_gb:.2f} GB です。")
     result_lines.append("")
     result_lines.append("推論及び、トレーニングに必要な概算のGPUメモリサイズは")
-    result_lines.append(f"【推論】約 {estimated_gb * 1.3:.2f} GB")
-    result_lines.append(f"【トレーニング（LoRA）】約 {estimated_gb * 1.8:.2f} GB")
-    result_lines.append(f"【トレーニング（フルパラメータ）】約 {estimated_gb * 4:.2f} GB")
+    result_lines.append(f"【推論】約 {estimated_gb * INF_MULT:.2f} GB")
+    result_lines.append(f"【トレーニング（LoRA）】約 {estimated_gb * LORA_MULT:.2f} GB")
+    result_lines.append(f"【トレーニング（フルパラメータ）】約 {estimated_gb * FULL_MULT:.2f} GB")
     result_lines.append("となります。")
 
     # GPU選定のコメント