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

Merge pull request #13 from UnderSampled/blip-download 

Directly download BLIP model to file
This commit is contained in:
Victor Hall 2022-11-25 14:06:38 -08:00 committed by GitHub
parent 8db26171cc 0d2119ea2f
commit 759228bf88
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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}")