Update set_ai_device.py

2f271d0e · xuwx1 · f8e1a2ec · 2f271d0e
Commit 2f271d0e authored Dec 11, 2025 by xuwx1
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

lightx2v_platform/set_ai_device.py lightx2v_platform/set_ai_device.py +1 -1

No files found.
--- a/lightx2v_platform/set_ai_device.py
+++ b/lightx2v_platform/set_ai_device.py
@@ -4,7 +4,7 @@ from lightx2v_platform import *


 def set_ai_device():
-    platform = os.getenv("PLATFORM", "cuda")
+    platform = "cuda"
    init_ai_device(platform)
    from lightx2v_platform.base.global_var import AI_DEVICE