Update app.py

app.py

@@ -7,8 +7,10 @@ import torch
 from PIL import Image
 import random
 import time
-from diffusers import HiDreamImagePipeline, HiDreamImageTransformer2DModel
-from diffusers.schedulers.flash_flow_match import FlashFlowMatchEulerDiscreteScheduler
+from hi_diffusers import HiDreamImagePipeline
+from hi_diffusers import HiDreamImageTransformer2DModel
+from hi_diffusers.schedulers.fm_solvers_unipc import FlowUniPCMultistepScheduler
+from hi_diffusers.schedulers.flash_flow_match import FlashFlowMatchEulerDiscreteScheduler
 from transformers import LlamaForCausalLM, PreTrainedTokenizerFast
 from huggingface_hub import ModelCard