Remove ZeroGPU smoke test
Browse files- 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])
|