mirror of
https://github.com/THUDM/CogVideo.git
synced 2025-04-06 03:57:56 +08:00
fix MODEL_LICENSE
This commit is contained in:
parent
71399f7558
commit
54546d0f89
@ -80,7 +80,7 @@ def generate_video(
|
|||||||
|
|
||||||
# Generate the video frames using the pipeline
|
# Generate the video frames using the pipeline
|
||||||
video = pipe(
|
video = pipe(
|
||||||
num_inference_steps=5, # Number of inference steps
|
num_inference_steps=num_inference_steps, # Number of inference steps
|
||||||
guidance_scale=guidance_scale, # Guidance scale for classifier-free guidance
|
guidance_scale=guidance_scale, # Guidance scale for classifier-free guidance
|
||||||
prompt_embeds=prompt_embeds, # Encoded prompt embeddings
|
prompt_embeds=prompt_embeds, # Encoded prompt embeddings
|
||||||
negative_prompt_embeds=torch.zeros_like(prompt_embeds), # Not Supported negative prompt
|
negative_prompt_embeds=torch.zeros_like(prompt_embeds), # Not Supported negative prompt
|
||||||
|
Loading…
x
Reference in New Issue
Block a user