Hugging Face's logo

Spaces:
Ryukijano
/
Cosmos_Sentinel
Running on Zero

App Files Community
Ryukijano commited on
Commit
d01f895
·
1 Parent(s): dc9e6df

fix: Catch missing cosmos_predict2 exception during warmup properly

Browse files
Files changed (1) hide show
  1. space_backend.py +0 -1
space_backend.py CHANGED
@@ -89,7 +89,6 @@ def preload_runtime(preload_badas=True, preload_reason=True, preload_predict=Fal
89
  get_reason_model_bundle(reason_model_name)
90
  steps.append(f"Reason model ready: {reason_model_name}")
91
  if preload_predict:
92
- get_predict_inference(predict_model_name, str(PREDICT_OUTPUT_ROOT), True)
93
  try:
94
  get_predict_inference(predict_model_name, str(PREDICT_OUTPUT_ROOT), True)
95
  steps.append(f"Predict model ready: {predict_model_name}")
 
89
  get_reason_model_bundle(reason_model_name)
90
  steps.append(f"Reason model ready: {reason_model_name}")
91
  if preload_predict:
 
92
  try:
93
  get_predict_inference(predict_model_name, str(PREDICT_OUTPUT_ROOT), True)
94
  steps.append(f"Predict model ready: {predict_model_name}")