fffiloni commited on
Commit
04c3d70
·
verified ·
1 Parent(s): 8c83f7a

skip load if not on gpu instance

Browse files
Files changed (1) hide show
  1. gradio/vace.py +8 -1
gradio/vace.py CHANGED
@@ -105,6 +105,7 @@ class VACEInference:
105
  self.save_dir = cfg.save_dir
106
  self.gallery_share = gallery_share
107
  self.gallery_share_data = FixedSizeQueue(max_size=gallery_share_limit)
 
108
  if not skip_load:
109
  if not args.mp:
110
  self.pipe = WanVace(
@@ -416,7 +417,13 @@ if __name__ == '__main__':
416
  os.makedirs(args.save_dir, exist_ok=True)
417
 
418
  with gr.Blocks(css=css) as demo:
419
- infer_gr = VACEInference(args, skip_load=False, gallery_share=True, gallery_share_limit=5)
 
 
 
 
 
 
420
  infer_gr.create_ui()
421
  infer_gr.set_callbacks()
422
  allowed_paths = [args.save_dir]
 
105
  self.save_dir = cfg.save_dir
106
  self.gallery_share = gallery_share
107
  self.gallery_share_data = FixedSizeQueue(max_size=gallery_share_limit)
108
+
109
  if not skip_load:
110
  if not args.mp:
111
  self.pipe = WanVace(
 
417
  os.makedirs(args.save_dir, exist_ok=True)
418
 
419
  with gr.Blocks(css=css) as demo:
420
+
421
+ if not is_shared_ui and is_gpu_associated:
422
+ skip_load = True
423
+ else:
424
+ skip_load = False
425
+
426
+ infer_gr = VACEInference(args, skip_load=skip_load, gallery_share=True, gallery_share_limit=5)
427
  infer_gr.create_ui()
428
  infer_gr.set_callbacks()
429
  allowed_paths = [args.save_dir]