up

app-fast.py

@@ -63,13 +63,15 @@ pipe = HiDreamImagePipeline.from_pretrained(
 pipe.transformer = transformer
 
 
-@spaces.GPU(duration=60 * 3)
+@spaces.GPU(duration=120)
 def generate_image(
     prompt: str, resolution: str, seed: int, progress=gr.Progress(track_tqdm=True)
 ) -> tuple[PIL.Image.Image, int]:
     if seed == -1:
         seed = torch.randint(0, 1_000_000, (1,)).item()
 
+    msg = "ℹ️ This spaces currently crash because of the memory usage. Please help me fix 😅"
+    raise gr.Error(msg, duration=10)
     height, width = tuple(map(int, resolution.replace(" ", "").split("x")))
     generator = torch.Generator("cuda").manual_seed(seed)
 
@@ -87,8 +89,8 @@ def generate_image(
 
 
 # Gradio UI
-with gr.Blocks(title="HiDream Image Generator") as demo:
-    gr.Markdown("## 🌈 HiDream Image Generator")
+with gr.Blocks(title="HiDream Image Generator Fast") as demo:
+    gr.Markdown("## 🌈 HiDream Image Generator Fast")
 
     with gr.Row():
         with gr.Column():