diff --git a/scripts/core/txt2vid.py b/scripts/core/txt2vid.py
index a03784a..536e0a9 100644
--- a/scripts/core/txt2vid.py
+++ b/scripts/core/txt2vid.py
@@ -6,7 +6,7 @@ import numpy as np
 from PIL import Image
 
 import modules.paths as ph
-from modules.shared import devices
+from modules import devices
 
 from scripts.core import utils, flow_utils
 from FloweR.model import FloweR
@@ -237,4 +237,4 @@ def start_process(*args):
 
     # print('TOTAL TIME:', int(time.time() - processing_start_time))
 
-    yield 'done', curr_frame, occlusion_mask, warped_styled_frame_, processed_frame, output_video_name, gr.Button.update(interactive=True), gr.Button.update(interactive=False)
\ No newline at end of file
+    yield 'done', curr_frame, occlusion_mask, warped_styled_frame_, processed_frame, output_video_name, gr.Button.update(interactive=True), gr.Button.update(interactive=False)