memory optimization for CLIP interrogator

changed default cfg_scale to a higher value
This commit is contained in:
AUTOMATIC
2022-09-12 11:55:27 +03:00
parent ab0a79cdf4
commit 9bb20be090
4 changed files with 36 additions and 7 deletions

View File

@@ -5,6 +5,16 @@ module_in_gpu = None
cpu = torch.device("cpu")
device = gpu = get_optimal_device()
def send_everything_to_cpu():
global module_in_gpu
if module_in_gpu is not None:
module_in_gpu.to(cpu)
module_in_gpu = None
def setup_for_low_vram(sd_model, use_medvram):
parents = {}