Update app.py

2026-01-19 01:48:46 +00:00
parent 06a6fb0312
commit 6c17d7721a
1 changed files with 13 additions and 1 deletions
--- a/app.py
+++ b/app.py
@@ -4,8 +4,20 @@ import random
 import torch
 import spaces

+
+
+
+import gc
+
+from safetensors.torch import load_file
+from huggingface_hub import hf_hub_download
+
+
+
+
+
 from PIL import Image
-from diffusers import FlowMatchEulerDiscreteScheduler, QwenImageEditPlusPipeline
+from diffusers import FlowMatchEulerDiscreteScheduler, QwenImageEditPlusPipeline, EulerAncestralDiscreteScheduler, FlowMatchEulerDiscreteScheduler
 # from optimization import optimize_pipeline_
 # from qwenimage.pipeline_qwenimage_edit_plus import QwenImageEditPlusPipeline
 # from qwenimage.transformer_qwenimage import QwenImageTransformer2DModel