Zhengrui commited on
Commit
d61df16
·
verified ·
1 Parent(s): 90b3cc9

Remove ZeroGPU smoke test

Browse files
Files changed (1) hide show
  1. app_dvd_image.py +0 -17
app_dvd_image.py CHANGED
@@ -457,18 +457,6 @@ def worker_path(name: str) -> str:
457
  return str(user_dir / name)
458
 
459
 
460
- @spaces.GPU(duration=30)
461
- def zero_gpu_smoke_test():
462
- log_event("zero_gpu_smoke_test start")
463
- if not torch.cuda.is_available():
464
- log_event("zero_gpu_smoke_test no cuda")
465
- return "CUDA unavailable inside ZeroGPU worker"
466
- value = torch.ones((1,), device="cuda").sum().item()
467
- name = torch.cuda.get_device_name(0)
468
- log_event(f"zero_gpu_smoke_test done device={name} value={value}")
469
- return f"OK: {name}, value={value}"
470
-
471
-
472
  def get_seed(randomize_seed: bool, seed: int) -> int:
473
  return int(np.random.randint(0, MAX_SEED)) if randomize_seed else int(seed)
474
 
@@ -907,10 +895,6 @@ with gr.Blocks(
907
  """
908
  )
909
 
910
- with gr.Row():
911
- zero_gpu_smoke_btn = gr.Button("ZeroGPU Smoke Test")
912
- zero_gpu_smoke_out = gr.Textbox(label="ZeroGPU Status", interactive=False)
913
-
914
  generated_voxels_state = gr.State()
915
  edit_voxels_state = gr.State()
916
  edited_voxels_state = gr.State()
@@ -1057,7 +1041,6 @@ with gr.Blocks(
1057
  demo.load(start_session)
1058
  demo.unload(end_session)
1059
 
1060
- zero_gpu_smoke_btn.click(zero_gpu_smoke_test, outputs=[zero_gpu_smoke_out])
1061
 
1062
  gen_example.change(load_generation_example, inputs=[gen_example], outputs=[gen_image])
1063
  edit_image_example.change(load_edit_image_example, inputs=[edit_image_example], outputs=[edit_target_image])
 
457
  return str(user_dir / name)
458
 
459
 
 
 
 
 
 
 
 
 
 
 
 
 
460
  def get_seed(randomize_seed: bool, seed: int) -> int:
461
  return int(np.random.randint(0, MAX_SEED)) if randomize_seed else int(seed)
462
 
 
895
  """
896
  )
897
 
 
 
 
 
898
  generated_voxels_state = gr.State()
899
  edit_voxels_state = gr.State()
900
  edited_voxels_state = gr.State()
 
1041
  demo.load(start_session)
1042
  demo.unload(end_session)
1043
 
 
1044
 
1045
  gen_example.change(load_generation_example, inputs=[gen_example], outputs=[gen_image])
1046
  edit_image_example.change(load_edit_image_example, inputs=[edit_image_example], outputs=[edit_target_image])