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

Change low VRAM threshold to include 12GB gpus

Browse Source
This commit is contained in:
pharmapsychotic
2023-03-22 13:16:21 -05:00
parent 04208d88fb
commit 0f9a6d1a93
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/clip_interrogator_ext.py
View File

@@ -11,7 +11,7 @@ from clip_interrogator import Config, Interrogator, list_caption_models, list_cl
from modules import devices, lowvram, script_callbacks, shared
__version__ = '0.1.2'
__version__ = '0.1.3'
ci = None
low_vram = False
@@ -275,7 +275,7 @@ def add_tab():
if not low_vram and torch.cuda.is_available():
device = devices.get_optimal_device()
vram_total = torch.cuda.get_device_properties(device).total_memory
if vram_total < 11*1024*1024*1024:
if vram_total <= 12*1024*1024*1024:
low_vram = True
with gr.Blocks(analytics_enabled=False) as ui: