File size: 263 Bytes
27435c3
 
43afb20
27435c3
43afb20
27435c3
43afb20
27435c3
43afb20
 
 
1
2
3
4
5
6
7
8
9
10
11
from diffusers import DiffusionPipeline

pipeline=DiffusionPipeline.from_pretrained("THUDM/CogVideoX-2b")

def generate_output()

output = pipeline("your input prompt here")return output

if __name__ == "__main__":
    result = generate_output()
    print(result)