Hugging Face's logo

Spaces:
hlmshkr
/
gpu-sizing-calculator
Sleeping

App Files Community
hlmshkr commited on
Commit
5770769
·
1 Parent(s): f6a93d1

deploy

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -106,6 +106,7 @@ def _predict_generic(context, TP, DP, concurrency, gpu_type, coeffs):
106
 
107
  def predict_latency(model, context, TP, DP, concurrency, gpu_type):
108
  if TP * DP > 8:
 
109
  msg = "⚠️ Invalid configuration: TP × DP must not exceed 8 GPUs."
110
  return (msg, msg)
111
  if gpu_type not in ["8× H100", "8× H200"]:
 
106
 
107
  def predict_latency(model, context, TP, DP, concurrency, gpu_type):
108
  if TP * DP > 8:
109
+ print("")
110
  msg = "⚠️ Invalid configuration: TP × DP must not exceed 8 GPUs."
111
  return (msg, msg)
112
  if gpu_type not in ["8× H100", "8× H200"]: