Update README.md

README.md

@@ -62,6 +62,9 @@ pip install transformers diffusers
 from diffusers.utils import export_to_video
 import tqdm
 from torchvision.transforms import ToPILImage
+import torch
+from transformers import AutoTokenizer, AutoModelForCausalLM
+from diffusers import CogVideoXTransformer3DModel, AutoencoderKLCogVideoX
 
 prompt="チューリップや菜の花、色とりどりの花が果てしなく続く畑を埋め尽くし、まるでパッチワークのようにカラフルに彩る。朝の柔らかな光が花びらを透かし、淡いグラデーションが映える。風に揺れる花々をスローモーションで捉え、花びらが優雅に舞う姿を映画のような演出で撮影。背景には遠くに連なる山並みや青い空、浮かぶ白い雲が立体感を引き立てる。"
 device="cuda"