JLNTV_IPD/CogVideo
4
0
Fork 0
mirror of https://github.com/THUDM/CogVideo.git synced 2025-12-02 10:32:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update predict.py

Browse Source
This commit is contained in:
Stephan Auerhahn 2022-07-26 22:58:19 -07:00 committed by GitHub
parent 7e5cd59fd0
commit e0067c5132
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
predict.py
View File

@ -497,7 +497,8 @@ class Predictor(BasePredictor):
self.args.seed = seed
self.args.use_guidance_stage1 = use_guidance
self.prompt = prompt
if os.path.exists(image_prompt):
self.image_prompt = None
if os.path.exists(str(image_prompt)):
try:
Image.open(str(image_prompt))
except (FileNotFoundError, UnidentifiedImageError):