Spaces:
Runtime error
Runtime error
Update wan/image2video_if_oss.py
Browse files
wan/image2video_if_oss.py
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
# Copyright 2024-2025 The Alibaba Wan Team Authors. All rights reserved.
|
| 2 |
-
import sys
|
| 3 |
sys.path.append('../OSS')
|
| 4 |
from OSS.OSS import search_OSS_video, infer_OSS
|
| 5 |
from OSS.model_wrap import _WrappedModel_Wan
|
|
@@ -130,6 +130,7 @@ class WanI2V:
|
|
| 130 |
|
| 131 |
self.sample_neg_prompt = config.sample_neg_prompt
|
| 132 |
|
|
|
|
| 133 |
def generate(self,
|
| 134 |
input_prompt,
|
| 135 |
img,
|
|
|
|
| 1 |
# Copyright 2024-2025 The Alibaba Wan Team Authors. All rights reserved.
|
| 2 |
+
import sys,spaces
|
| 3 |
sys.path.append('../OSS')
|
| 4 |
from OSS.OSS import search_OSS_video, infer_OSS
|
| 5 |
from OSS.model_wrap import _WrappedModel_Wan
|
|
|
|
| 130 |
|
| 131 |
self.sample_neg_prompt = config.sample_neg_prompt
|
| 132 |
|
| 133 |
+
@spaces.GPU
|
| 134 |
def generate(self,
|
| 135 |
input_prompt,
|
| 136 |
img,
|