pharmapsychotic/clip-interrogator-ext
1
0
Fork 0
mirror of https://github.com/pharmapsychotic/clip-interrogator-ext.git synced 2026-04-30 19:21:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

.

Browse Source
This commit is contained in:
pharmapsychotic
2023-02-18 23:04:04 -06:00
parent 31a3a440d3
commit 72cf1ea1c1
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/clip_interrogator_ext.py
View File

@@ -25,8 +25,8 @@ def load(clip_model_name):
print(f" detected < 12GB VRAM, using low VRAM mode") print(f" detected < 12GB VRAM, using low VRAM mode")
config = Config(device=devices.get_optimal_device(), clip_model_name=clip_model_name) config = Config(device=devices.get_optimal_device(), clip_model_name=clip_model_name)
config.cache_path = 'models/clip-interrogator'
if low_vram: if low_vram:
config.cache_path = 'models/clip-interrogator'
config.blip_model_type = 'base' config.blip_model_type = 'base'
config.blip_offload = True config.blip_offload = True
config.chunk_size = 1024 config.chunk_size = 1024