remove reundant stream=true

This commit is contained in:
jmorganca 2026-01-22 22:37:36 -08:00
parent 9611be43d6
commit 23a46bf854

View File

@ -7,12 +7,12 @@ from ollama import generate
prompt = 'a sunset over mountains'
print(f'Prompt: {prompt}')
for response in generate(model='x/z-image-turbo', prompt=prompt, stream=True):
for response in generate(model='x/z-image-turbo', prompt=prompt):
if response.image:
# Final response contains the image
with open('output.png', 'wb') as f:
f.write(base64.b64decode(response.image))
print('\nImage saved to output.png')
else:
elif response.total:
# Progress update
print(f'Progress: {response.completed}/{response.total}', end='\r')