diff --git a/iic.py b/iic.py
index 0885768..b968356 100644
--- a/iic.py
+++ b/iic.py
@@ -1,11 +1,11 @@
 import os
 import torch
+device = "cuda" if torch.cuda.is_available() else "npu"
 import patch
 
 from modelscope.pipelines import pipeline
 from modelscope.outputs import OutputKeys
 
-device = "cuda" if torch.cuda.is_available() else "npu"
 model_path = "/mnt/contest_ceph/zhanghao/models/iic/text-to-video-synthesis"
 p = pipeline('text-to-video-synthesis', model_path, device=device)