1.0.0
Warning
Breaking changes
This 1.0.0 latest release of replicate
contains breaking changes. The replicate.run()
method will now return FileObjects
rather than URL strings by default for models that output files.
The FileObject
implements an iterable object similar to httpx.Response
to make it easier to work with files and ensures that Replicate can deliver file data to the client in the most efficient manner possible.
For example:
[output] = replicate.run(
"black-forest-labs/flux-schnell",
input={"prompt": "astronaut riding a rocket like a horse"},
);
For most basic cases you'll want to utilize either the url
or read()
fields depending on whether you want to directly consume the file or pass it on.
To access the file URL:
print(output.url) #=> "https://delivery.replicate.com/..."
To consume the file directly:
with open('output.bin', 'wb') as file:
file.write(output.read())
Or for very large files they can be streamed:
with open(file_path, 'wb') as file:
for chunk in output:
file.write(chunk)
Each of these methods has an equivalent asyncio
API.
async with aiofiles.open(filename, 'w') as file:
await file.write(await output.aread())
async with aiofiles.open(filename, 'w') as file:
await for chunk in output:
await file.write(chunk)
For streaming responses from common frameworks, all support taking Iterator
types:
Django
@condition(etag_func=None)
def stream_response(request):
output = replicate.run("black-forest-labs/flux-schnell", input={...}, use_file_output =True)
return HttpResponse(output, content_type='image/webp')
FastAPI
@app.get("/")
async def main():
output = replicate.run("black-forest-labs/flux-schnell", input={...}, use_file_output =True)
return StreamingResponse(output)
Flask
@app.route('/stream')
def streamed_response():
output = replicate.run("black-forest-labs/flux-schnell", input={...}, use_file_output =True)
return app.response_class(stream_with_context(output))
In case of breakages, in most instances, updating existing applications to call output.url
should fix issues.
To revert to previous behaviour you can opt out of FileOutput
by passing use_file_output=False
to the replicate.run()
method.
const replicate = replicate.run("acmecorp/acme-model", use_file_output=False);
Full Changelog: 0.34.2...1.0.0