diff --git a/app.py b/app.py index 8958b2c..5be6b9e 100644 --- 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