fffiloni commited on
Commit
84cf951
·
verified ·
1 Parent(s): 2307c3a

import wan only if gpu avalaible

Browse files
Files changed (1) hide show
  1. gradio/vace.py +4 -4
gradio/vace.py CHANGED
@@ -21,10 +21,10 @@ if not is_shared_ui and is_gpu_associated:
21
  local_dir = "./models/Wan2.1-VACE-1.3B"
22
  )
23
 
24
- sys.path.insert(0, os.path.sep.join(os.path.realpath(__file__).split(os.path.sep)[:-2]))
25
- import wan
26
- from wan import WanVace, WanVaceMP
27
- from wan.configs import WAN_CONFIGS, SIZE_CONFIGS
28
 
29
  css = """
30
  div#warning-duplicate {
 
21
  local_dir = "./models/Wan2.1-VACE-1.3B"
22
  )
23
 
24
+ sys.path.insert(0, os.path.sep.join(os.path.realpath(__file__).split(os.path.sep)[:-2]))
25
+ import wan
26
+ from wan import WanVace, WanVaceMP
27
+ from wan.configs import WAN_CONFIGS, SIZE_CONFIGS
28
 
29
  css = """
30
  div#warning-duplicate {