Decouple deterministic smoke from GPU quota
Browse files
app.py
CHANGED
|
@@ -633,8 +633,7 @@ def recent_failure_reports():
|
|
| 633 |
return [str(path) for path in sorted(OUTPUT_DIR.glob("*_failure.json"), key=lambda item: item.stat().st_mtime, reverse=True)[:10]]
|
| 634 |
|
| 635 |
|
| 636 |
-
|
| 637 |
-
def generate(image: Image.Image, extra_files, backend: str, guess: str, demo_mode: bool, progress=gr.Progress()):
|
| 638 |
if image is None:
|
| 639 |
raise gr.Error("Please add an input image first.")
|
| 640 |
started = time.perf_counter()
|
|
@@ -840,6 +839,18 @@ def generate(image: Image.Image, extra_files, backend: str, guess: str, demo_mod
|
|
| 840 |
) from exc
|
| 841 |
|
| 842 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 843 |
with gr.Blocks(title="Simam3D") as demo:
|
| 844 |
gr.Markdown("# Simam3D\nSingle-image depth and 3D reconstruction workbench")
|
| 845 |
gr.Markdown(
|
|
|
|
| 633 |
return [str(path) for path in sorted(OUTPUT_DIR.glob("*_failure.json"), key=lambda item: item.stat().st_mtime, reverse=True)[:10]]
|
| 634 |
|
| 635 |
|
| 636 |
+
def _generate_impl(image: Image.Image, extra_files, backend: str, guess: str, demo_mode: bool, progress=gr.Progress()):
|
|
|
|
| 637 |
if image is None:
|
| 638 |
raise gr.Error("Please add an input image first.")
|
| 639 |
started = time.perf_counter()
|
|
|
|
| 839 |
) from exc
|
| 840 |
|
| 841 |
|
| 842 |
+
@GPU_FUNCTION
|
| 843 |
+
def _generate_gpu(image: Image.Image, extra_files, backend: str, guess: str, demo_mode: bool, progress=gr.Progress()):
|
| 844 |
+
return _generate_impl(image, extra_files, backend, guess, demo_mode, progress)
|
| 845 |
+
|
| 846 |
+
|
| 847 |
+
def generate(image: Image.Image, extra_files, backend: str, guess: str, demo_mode: bool, progress=gr.Progress()):
|
| 848 |
+
"""Keep the deterministic export diagnostic independent of GPU quota."""
|
| 849 |
+
if demo_mode:
|
| 850 |
+
return _generate_impl(image, extra_files, backend, guess, demo_mode, progress)
|
| 851 |
+
return _generate_gpu(image, extra_files, backend, guess, demo_mode, progress)
|
| 852 |
+
|
| 853 |
+
|
| 854 |
with gr.Blocks(title="Simam3D") as demo:
|
| 855 |
gr.Markdown("# Simam3D\nSingle-image depth and 3D reconstruction workbench")
|
| 856 |
gr.Markdown(
|