diff --git a/apg_guidance.py b/acestep/apg_guidance.py
similarity index 100%
rename from apg_guidance.py
rename to acestep/apg_guidance.py
diff --git a/data_sampler.py b/acestep/data_sampler.py
similarity index 100%
rename from data_sampler.py
rename to acestep/data_sampler.py
diff --git a/app.py b/acestep/gui.py
similarity index 100%
rename from app.py
rename to acestep/gui.py
diff --git a/language_segmentation/LangSegment.py b/acestep/language_segmentation/LangSegment.py
similarity index 100%
rename from language_segmentation/LangSegment.py
rename to acestep/language_segmentation/LangSegment.py
diff --git a/language_segmentation/__init__.py b/acestep/language_segmentation/__init__.py
similarity index 100%
rename from language_segmentation/__init__.py
rename to acestep/language_segmentation/__init__.py
diff --git a/language_segmentation/utils/__init__.py b/acestep/language_segmentation/utils/__init__.py
similarity index 100%
rename from language_segmentation/utils/__init__.py
rename to acestep/language_segmentation/utils/__init__.py
diff --git a/language_segmentation/utils/num.py b/acestep/language_segmentation/utils/num.py
similarity index 100%
rename from language_segmentation/utils/num.py
rename to acestep/language_segmentation/utils/num.py
diff --git a/models/ace_step_transformer.py b/acestep/models/ace_step_transformer.py
similarity index 100%
rename from models/ace_step_transformer.py
rename to acestep/models/ace_step_transformer.py
diff --git a/models/attention.py b/acestep/models/attention.py
similarity index 100%
rename from models/attention.py
rename to acestep/models/attention.py
diff --git a/models/config.json b/acestep/models/config.json
similarity index 100%
rename from models/config.json
rename to acestep/models/config.json
diff --git a/models/customer_attention_processor.py b/acestep/models/customer_attention_processor.py
similarity index 100%
rename from models/customer_attention_processor.py
rename to acestep/models/customer_attention_processor.py
diff --git a/models/lyrics_utils/lyric_encoder.py b/acestep/models/lyrics_utils/lyric_encoder.py
similarity index 100%
rename from models/lyrics_utils/lyric_encoder.py
rename to acestep/models/lyrics_utils/lyric_encoder.py
diff --git a/models/lyrics_utils/lyric_normalizer.py b/acestep/models/lyrics_utils/lyric_normalizer.py
similarity index 100%
rename from models/lyrics_utils/lyric_normalizer.py
rename to acestep/models/lyrics_utils/lyric_normalizer.py
diff --git a/models/lyrics_utils/lyric_tokenizer.py b/acestep/models/lyrics_utils/lyric_tokenizer.py
similarity index 100%
rename from models/lyrics_utils/lyric_tokenizer.py
rename to acestep/models/lyrics_utils/lyric_tokenizer.py
diff --git a/models/lyrics_utils/vocab.json b/acestep/models/lyrics_utils/vocab.json
similarity index 100%
rename from models/lyrics_utils/vocab.json
rename to acestep/models/lyrics_utils/vocab.json
diff --git a/models/lyrics_utils/zh_num2words.py b/acestep/models/lyrics_utils/zh_num2words.py
similarity index 100%
rename from models/lyrics_utils/zh_num2words.py
rename to acestep/models/lyrics_utils/zh_num2words.py
diff --git a/music_dcae/__init__.py b/acestep/music_dcae/__init__.py
similarity index 100%
rename from music_dcae/__init__.py
rename to acestep/music_dcae/__init__.py
diff --git a/music_dcae/music_dcae_pipeline.py b/acestep/music_dcae/music_dcae_pipeline.py
similarity index 100%
rename from music_dcae/music_dcae_pipeline.py
rename to acestep/music_dcae/music_dcae_pipeline.py
diff --git a/music_dcae/music_log_mel.py b/acestep/music_dcae/music_log_mel.py
similarity index 100%
rename from music_dcae/music_log_mel.py
rename to acestep/music_dcae/music_log_mel.py
diff --git a/music_dcae/music_vocoder.py b/acestep/music_dcae/music_vocoder.py
similarity index 100%
rename from music_dcae/music_vocoder.py
rename to acestep/music_dcae/music_vocoder.py
diff --git a/pipeline_ace_step.py b/acestep/pipeline_ace_step.py
similarity index 100%
rename from pipeline_ace_step.py
rename to acestep/pipeline_ace_step.py
diff --git a/schedulers/scheduling_flow_match_euler_discrete.py b/acestep/schedulers/scheduling_flow_match_euler_discrete.py
similarity index 100%
rename from schedulers/scheduling_flow_match_euler_discrete.py
rename to acestep/schedulers/scheduling_flow_match_euler_discrete.py
diff --git a/schedulers/scheduling_flow_match_heun_discrete.py b/acestep/schedulers/scheduling_flow_match_heun_discrete.py
similarity index 100%
rename from schedulers/scheduling_flow_match_heun_discrete.py
rename to acestep/schedulers/scheduling_flow_match_heun_discrete.py
diff --git a/text2music_dataset.py b/acestep/text2music_dataset.py
similarity index 100%
rename from text2music_dataset.py
rename to acestep/text2music_dataset.py
diff --git a/ui/components.py b/acestep/ui/components.py
similarity index 100%
rename from ui/components.py
rename to acestep/ui/components.py