Update app.py

app.py

@@ -36,24 +36,33 @@ MAX_DURATION = round(MAX_FRAMES_MODEL / FIXED_FPS, 1)
 # LOAD PIPELINE
 # =========================================================
 print("Loading pipeline...")
+
+# 1. تحميل المكونات بدون تحديد device_map="cuda" لتجنب تعارض ZeroGPU
+transformer = WanTransformer3DModel.from_pretrained(
+    MODEL_ID,
+    subfolder="transformer",
+    torch_dtype=torch.bfloat16,
+    token=HF_TOKEN
+)
+
+transformer_2 = WanTransformer3DModel.from_pretrained(
+    MODEL_ID,
+    subfolder="transformer_2",
+    torch_dtype=torch.bfloat16,
+    token=HF_TOKEN
+)
+
+# 2. تجميع البايبلاين
 pipe = WanImageToVideoPipeline.from_pretrained(
     MODEL_ID,
-    transformer=WanTransformer3DModel.from_pretrained(
-        MODEL_ID,
-        subfolder="transformer",
-        torch_dtype=torch.bfloat16,
-        device_map="cuda",
-        token=HF_TOKEN
-    ),
-    transformer_2=WanTransformer3DModel.from_pretrained(
-        MODEL_ID,
-        subfolder="transformer_2",
-        torch_dtype=torch.bfloat16,
-        device_map="cuda",
-        token=HF_TOKEN
-    ),
+    transformer=transformer,
+    transformer_2=transformer_2,
     torch_dtype=torch.bfloat16,
-).to("cuda")
+)
+
+# 3. نقل الموديل للـ CUDA مرة واحدة هنا
+print("Moving pipeline to CUDA...")
+pipe = pipe.to("cuda")
 
 # =========================================================
 # LOAD LORA ADAPTERS
@@ -80,6 +89,7 @@ pipe.unload_lora_weights()
 # QUANTIZATION & AOT OPTIMIZATION
 # =========================================================
 print("Applying quantization...")
+# نقلنا التكميم بعد النقل للـ GPU لضمان التوافق
 quantize_(pipe.text_encoder, Int8WeightOnlyConfig())
 quantize_(pipe.transformer, Float8DynamicActivationFloat8WeightConfig())
 quantize_(pipe.transformer_2, Float8DynamicActivationFloat8WeightConfig())
@@ -275,19 +285,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
     generate_button.click(fn=generate_video, inputs=ui_inputs, outputs=[video_output, seed_input])
 
-    gr.Examples(
-        examples=[
-            [
-                "wan_i2v_input.JPG",
-                "POV selfie video, white cat with sunglasses standing on surfboard, relaxed smile, tropical beach behind (clear water, green hills, blue sky with clouds). Surfboard tips, cat falls into ocean, camera plunges underwater with bubbles and sunlight beams. Brief underwater view of cat’s face, then cat resurfaces, still filming selfie, playful summer vacation mood.",
-                4,
-            ],
-        ],
-        inputs=[input_image_component, prompt_input, steps_slider],
-        outputs=[video_output, seed_input],
-        fn=generate_video,
-        cache_examples=False 
-    )
+    # ملاحظة: يمكنك حذف جزء Examples إذا استمرت الأخطاء أو إذا لم تكن الصورة مرفوعة
+    # gr.Examples(...)
 
     # --- BOTTOM ADVERTISEMENT BANNER ---
     gr.HTML("""