Rename app.py to from transformers import AutoModelForCausalLM, AutoTokenizer model_name = "THUDM/CogVideoX-5B" tokenizer = AutoTokenizer.from_pretrained(model_name) model = AutoModelForCausalLM.from_pretrained(model_name)

#1
Ready to merge
This branch is ready to get merged automatically.
Your need to confirm your account before you can post a new comment.

Sign up or log in to comment