Don't download BLIP model in to memory before saving; Download directly to file.

This commit is contained in:
UnderSampled 2022-11-25 16:24:02 -05:00
parent 8db26171cc
commit 0d2119ea2f
1 changed files with 2 additions and 2 deletions

View File

@ -109,9 +109,9 @@ async def main(opt):
async with aiohttp.ClientSession() as session:
async with session.get(BLIP_MODEL_URL) as res:
result = await res.read()
with open(model_cache_path, 'wb') as f:
f.write(result)
async for chunk in res.content.iter_chunked(1024):
f.write(chunk)
print(f"Model cached to: {model_cache_path}")
else:
print(f"Model already cached to: {model_cache_path}")