[F] Fix import

2025-04-08 21:46:32 +00:00
parent 854a88f0ab
commit 0efca9dd1a
1 changed files with 5 additions and 4 deletions
@@ -1,11 +1,12 @@
 import torch
 import argparse
-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 . import HiDreamImagePipeline
+from . import HiDreamImageTransformer2DModel
+from .schedulers.fm_solvers_unipc import FlowUniPCMultistepScheduler
+from .schedulers.flash_flow_match import FlashFlowMatchEulerDiscreteScheduler
+

 MODEL_PREFIX = "azaneko"
 LLAMA_MODEL_NAME = "hugging-quants/Meta-Llama-3.1-8B-Instruct-GPTQ-INT4"