JLNTV_IPD/CogVideo
4
0
Fork 0
mirror of https://github.com/THUDM/CogVideo.git synced 2025-12-02 18:52:08 +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:29 -07:00 committed by GitHub
parent 634a513ba7
commit a054801c18
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
predict.py
View File

@ -504,7 +504,7 @@ class Predictor(BasePredictor):
except (FileNotFoundError, UnidentifiedImageError): except (FileNotFoundError, UnidentifiedImageError):
logging.debug("Bad image prompt; ignoring") # Is there a better way to input images? logging.debug("Bad image prompt; ignoring") # Is there a better way to input images?
else: else:
self.image_prompt = Image.open(image_prompt) self.image_prompt = str(image_prompt)
self.args.both_stages = both_stages self.args.both_stages = both_stages
for file in self.run(): for file in self.run():