From 0efca9dd1a9fc619eb1347620382bdc8912ef3c0 Mon Sep 17 00:00:00 2001
From: Azalea <22280294+hykilpikonna@users.noreply.github.com>
Date: Tue, 8 Apr 2025 21:46:32 +0000
Subject: [PATCH] [F] Fix import

---
 hi_diffusers/nf4.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/hi_diffusers/nf4.py b/hi_diffusers/nf4.py
index dfefa49..47df5dc 100644
--- a/hi_diffusers/nf4.py
+++ b/hi_diffusers/nf4.py
@@ -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"