JLNTV_IPD/CogVideo
4
0
Fork 0
mirror of https://github.com/THUDM/CogVideo.git synced 2025-04-05 19:41:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #58 from learningpro/main

Browse Source 
use huggingface model path and update requirements.txt to avoid PIL error
This commit is contained in:
zR 2024-08-06 15:55:25 +08:00 committed by GitHub
commit c3c708e028
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
gradio_demo.py
View File

@ -17,7 +17,7 @@ import PIL
dtype = torch.bfloat16
device = "cuda" if torch.cuda.is_available() else "cpu"
pipe = CogVideoXPipeline.from_pretrained("/share/home/zyx/Models/cogvideox-hf-0805", torch_dtype=dtype).to(device)
pipe = CogVideoXPipeline.from_pretrained("THUDM/CogVideoX-2b", torch_dtype=dtype).to(device)
sys_prompt = """You are part of a team of bots that creates videos. You work with an assistant bot that will draw anything you say in square brackets.

3
requirements.txt
View File

@ -6,4 +6,5 @@ opencv-python>=4.10
imageio-ffmpeg>=0.5.1
openai>=1.38.0
transformers>=4.43.3
accelerate>=0.33.0
accelerate>=0.33.0
pillow==9.5.0