Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2025-11-04 14:16:57 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Added missing requirements

Browse Source
This commit is contained in:
DeepBeepMeep 2025-04-09 22:11:04 +02:00
parent 016e57c253
commit eba935515a
2 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
requirements.txt
View File

@ -22,4 +22,5 @@ mutagen
decord decord
onnxruntime-gpu onnxruntime-gpu
rembg[gpu]==2.0.65 rembg[gpu]==2.0.65
matplotlib
# rembg==2.0.65 # rembg==2.0.65

13
wgp.py
View File

@ -173,9 +173,14 @@ def process_prompt_and_add_tasks(state, model_choice):
return return
else: else:
video_mask = None video_mask = None
if "O" in video_prompt_type and inputs["max_frames"]==0: if "O" in video_prompt_type :
gr.Info(f"In order to extend a video, you need to indicate how many frames you want to reuse in the source video.") max_frames= inputs["max_frames"]
return video_length = inputs["video_length"]
if max_frames ==0:
gr.Info(f"Warning : you have asked to reuse all the frames of the control Video before extending it. Please make sure the number of frames of the control Video is lower than the total number of frames to generate otherwise it won't make a difference.")
elif max_frames >= video_length:
gr.Info(f"The number of frames in the control Video to reuse ({max_frames}) before extending the Video can not be bigger than the total number of frames ({video_length}) to generate.")
return
if isinstance(image_refs, list): if isinstance(image_refs, list):
image_refs = [ convert_image(tup[0]) for tup in image_refs ] image_refs = [ convert_image(tup[0]) for tup in image_refs ]
@ -2060,7 +2065,7 @@ def generate_video(
# gr.Info("Unable to generate a Video while a new configuration is being applied.") # gr.Info("Unable to generate a Video while a new configuration is being applied.")
# return # return
if "P" in preload_model_policy: if "P" in preload_model_policy and not "U" in preload_model_policy:
while wan_model == None: while wan_model == None:
time.sleep(1) time.sleep(1)