mirror of
https://github.com/ollama/ollama-python.git
synced 2026-05-01 11:48:17 +08:00
remove reundant stream=true
This commit is contained in:
parent
9611be43d6
commit
23a46bf854
@ -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')
|
||||
|
||||
Loading…
Reference in New Issue
Block a user