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"