add stream=True

This commit is contained in:
jmorganca 2026-01-22 22:40:40 -08:00
parent 23a46bf854
commit 3b33dd3bf4

View File

@ -7,7 +7,7 @@ from ollama import generate
prompt = 'a sunset over mountains'
print(f'Prompt: {prompt}')
for response in generate(model='x/z-image-turbo', prompt=prompt):
for response in generate(model='x/z-image-turbo', prompt=prompt, stream=True):
if response.image:
# Final response contains the image
with open('output.png', 'wb') as f: