Comment

app.py

@@ -5,8 +5,8 @@ import requests
 # 定数定義
 API_BASE_URL = "https://huggingface.co/api/models/"
 MULTIPLIERS = {
-    "U8": 1,
     "I8": 1,
+    "U8": 1,
     "F8_E5M2": 1,
     "F8_E4M3": 1,
     "I16": 2,
@@ -16,11 +16,27 @@ MULTIPLIERS = {
     "I32": 4,
     "U32": 4,
     "F32": 4,
-    "F64": 8,
     "I64": 8,
     "U64": 8,
+    "F64": 8,
 }
 PRECISION_KEYS = list(MULTIPLIERS.keys())
+COMMENT = {
+    "I8": "整数なのでGPUでの処理に向いてないかもしれません。",
+    "U8": "整数なのでGPUでの処理に向いてないかもしれません。",
+    "F8_E5M2": "FP8の処理能力が高いGPUを選定してください。",
+    "F8_E4M3": "FP8の処理能力が高いGPUを選定してください。",
+    "I16": "整数なのでGPUでの処理に向いてないかもしれません。",
+    "U16": "整数なのでGPUでの処理に向いてないかもしれません。",
+    "F16": "FP16の処理能力が高いGPUを選定してください。",
+    "BF16": "BF16の処理能力が高いGPUを選定してください。",
+    "I32": "整数なのでGPUでの処理に向いてないかもしれません。",
+    "U32": "整数なのでGPUでの処理に向いてないかもしれません。",
+    "F32": "FP32の処理能力が高いGPUを選定してください。",
+    "I64": "整数なのでGPUでの処理に向いてないかもしれません。",
+    "U64": "整数なのでGPUでの処理に向いてないかもしれません。",
+    "F64": "FP64の処理能力が高いGPUを選定してください。",
+}
 
 def get_model_api_info(model_name: str) -> dict:
     """
@@ -92,7 +108,13 @@ def estimate_gpu_memory(model_name: str) -> str:
     result_lines.append(f"モデル '{model_name}' のsafetensors情報より、各パラメータ数は")
 
     # 各精度ごとのパラメータ数の出力
+    max_precision = None
+    max_count = 0
     for precision in PRECISION_KEYS:
+        count = parameters.get(precision, 0)
+        if count > max_count:
+            max_precision = precision
+            max_count = count
         if precision in parameters:
             result_lines.append(f"【{precision}】 {parameters[precision]:,}")
 
@@ -103,6 +125,11 @@ def estimate_gpu_memory(model_name: str) -> str:
     result_lines.append(f"【トレーニング】約 {estimated_gb * 4:.2f} GB")
     result_lines.append("となります。")
 
+    # GPU選定のコメント
+    if max_precision is not None:
+        comment_message = COMMENT.get(max_precision, "")
+        result_lines.append(comment_message)
+
     if usedStorage_gb is not None:
         result_lines.append("")
         result_lines.append(f"参考: 該当リポジトリのファイルサイズは合計 {usedStorage_gb:.2f} GB")