Connie-Wild commited on
Commit
08bf387
·
1 Parent(s): 3087a2a
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -94,7 +94,8 @@ def estimate_gpu_memory(model_name: str) -> str:
94
  if usedStorage_gb is not None:
95
  result_lines.append("")
96
  result_lines.append(f"参考までに、該当リポジトリのファイルサイズは合計 {usedStorage_gb:.2f} GBです。")
97
- result_lines.append(f"これを全てモデルのデータとして仮定した場合、推論及び、トレーニングに必要な概算のGPUメモリサイズは")
 
98
  result_lines.append(f"【推論】約 {usedStorage_gb * 1.5:.2f} GB")
99
  result_lines.append(f"【トレーニング(LoRA)】約 {usedStorage_gb * 2:.2f} GB")
100
  result_lines.append(f"【トレーニング(フルパラメータ)】約 {usedStorage_gb * 4:.2f} GB")
@@ -121,7 +122,7 @@ def estimate_gpu_memory(model_name: str) -> str:
121
 
122
  result_lines.append(f"これらを合算するとモデルのデータサイズは約 {estimated_gb:.2f} GB です。")
123
  result_lines.append("")
124
- result_lines.append(f"推論及び、トレーニングに必要な概算のGPUメモリサイズは")
125
  result_lines.append(f"【推論】約 {estimated_gb * 1.5:.2f} GB")
126
  result_lines.append(f"【トレーニング(LoRA)】約 {estimated_gb * 2:.2f} GB")
127
  result_lines.append(f"【トレーニング(フルパラメータ)】約 {estimated_gb * 4:.2f} GB")
 
94
  if usedStorage_gb is not None:
95
  result_lines.append("")
96
  result_lines.append(f"参考までに、該当リポジトリのファイルサイズは合計 {usedStorage_gb:.2f} GBです。")
97
+ result_lines.append("")
98
+ result_lines.append("これを全てモデルのデータとして仮定した場合、推論及び、トレーニングに必要な概算のGPUメモリサイズは")
99
  result_lines.append(f"【推論】約 {usedStorage_gb * 1.5:.2f} GB")
100
  result_lines.append(f"【トレーニング(LoRA)】約 {usedStorage_gb * 2:.2f} GB")
101
  result_lines.append(f"【トレーニング(フルパラメータ)】約 {usedStorage_gb * 4:.2f} GB")
 
122
 
123
  result_lines.append(f"これらを合算するとモデルのデータサイズは約 {estimated_gb:.2f} GB です。")
124
  result_lines.append("")
125
+ result_lines.append("推論及び、トレーニングに必要な概算のGPUメモリサイズは")
126
  result_lines.append(f"【推論】約 {estimated_gb * 1.5:.2f} GB")
127
  result_lines.append(f"【トレーニング(LoRA)】約 {estimated_gb * 2:.2f} GB")
128
  result_lines.append(f"【トレーニング(フルパラメータ)】約 {estimated_gb * 4:.2f} GB")