tanner/min-dalle-test
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

update predict.py

Browse Source
This commit is contained in:
Brett Kuprel 2022-07-04 18:41:00 -04:00
parent d3beb3ab57
commit cf5b116284
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
replicate/predict.py
View File

@ -32,7 +32,7 @@ class Predictor(BasePredictor):
), ),
) -> Path: ) -> Path:
def handle_intermediate_image(i: int, image: Image.Image) -> Path: def handle_intermediate_image(i: int, image: Image.Image):
if i + 1 == 16: return if i + 1 == 16: return
out_path = Path(tempfile.mkdtemp()) / 'output.jpg' out_path = Path(tempfile.mkdtemp()) / 'output.jpg'
image.save(str(out_path)) image.save(str(out_path))