Mirrors
/
EveryDream
mirror of https://github.com/victorchall/EveryDream.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/auto_caption.py
View File

@ -109,9 +109,9 @@ async def main(opt):
async with aiohttp.ClientSession() as session: async with aiohttp.ClientSession() as session:
async with session.get(BLIP_MODEL_URL) as res: async with session.get(BLIP_MODEL_URL) as res:
result = await res.read()
with open(model_cache_path, 'wb') as f: 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}") print(f"Model cached to: {model_cache_path}")
else: else:
print(f"Model already cached to: {model_cache_path}") print(f"Model already cached to: {model_cache_path}")