Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 17:24:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

preenable kolors

Browse Source
This commit is contained in:
Vladimir Mandic
2024-07-10 12:36:34 -04:00
parent 8e170f54a6
commit 112b1d8775
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/model_kolors.py
+3 -3
View File
@@ -2,12 +2,11 @@ import torch
import diffusers
repo_id = 'Kwai-Kolors/Kolors'
repo_id = 'Kwai-Kolors/Kolors-diffusers'
def load_kolors(_checkpoint_info, diffusers_load_config={}):
from modules import shared, devices, modelloader
modelloader.hf_login()
from modules import shared, devices
diffusers_load_config['variant'] = "fp16"
if 'torch_dtype' not in diffusers_load_config:
diffusers_load_config['torch_dtype'] = torch.float16
@@ -23,5 +22,6 @@ def load_kolors(_checkpoint_info, diffusers_load_config={}):
cache_dir = shared.opts.diffusers_dir,
**diffusers_load_config,
)
pipe.vae.config.force_upcast = True
devices.torch_gc()
return pipe