Spaces:
Running
Running
skip load if not on gpu instance
Browse files- 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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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]
|