Skip to content

GitLab

Commit 2f271d0e authored by xuwx1's avatar xuwx1
Browse files

Update set_ai_device.py

parent f8e1a2ec
Pipeline #3155 failed with stages
in 0 seconds
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
lightx2v_platform/set_ai_device.py
View file @ 2f271d0e
...@@ -4,7 +4,7 @@ from lightx2v_platform import * ...@@ -4,7 +4,7 @@ from lightx2v_platform import *
def set_ai_device(): def set_ai_device():
platform = os.getenv("PLATFORM", "cuda") platform = "cuda"
init_ai_device(platform) init_ai_device(platform)
from lightx2v_platform.base.global_var import AI_DEVICE from lightx2v_platform.base.global_var import AI_DEVICE
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment