From 292ccef7bf60ee5c6934c215958ef09911bbaa7f Mon Sep 17 00:00:00 2001
From: chuxij <junmingong@qq.com>
Date: Wed, 14 May 2025 03:37:07 +0000
Subject: [PATCH] asign cuda device

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

diff --git a/acestep/gui.py b/acestep/gui.py
index db50a5d..1165507 100644
--- a/acestep/gui.py
+++ b/acestep/gui.py
@@ -9,11 +9,6 @@ Apache 2.0 License
 import os
 import click
 
-from acestep.ui.components import create_main_demo_ui
-from acestep.pipeline_ace_step import ACEStepPipeline
-from acestep.data_sampler import DataSampler
-
-
 @click.command()
 @click.option(
     "--checkpoint_path",
@@ -59,6 +54,10 @@ def main(checkpoint_path, server_name, port, device_id, share, bf16, torch_compi
 
     os.environ["CUDA_VISIBLE_DEVICES"] = str(device_id)
 
+    from acestep.ui.components import create_main_demo_ui
+    from acestep.pipeline_ace_step import ACEStepPipeline
+    from acestep.data_sampler import DataSampler
+
     model_demo = ACEStepPipeline(
         checkpoint_dir=checkpoint_path,
         dtype="bfloat16" if bf16 else "float32",