diff --git a/html/reference.json b/html/reference.json index dfc0a59df..4d65aa594 100644 --- a/html/reference.json +++ b/html/reference.json @@ -198,7 +198,7 @@ }, "Kwai Kolors": { - "path": "Kwai-Kolors/Kolors", + "path": "Kwai-Kolors/Kolors-diffusers", "desc": "Kolors is a large-scale text-to-image generation model based on latent diffusion, developed by the Kuaishou Kolors team. Trained on billions of text-image pairs, Kolors exhibits significant advantages over both open-source and proprietary models in visual quality, complex semantic accuracy, and text rendering for both Chinese and English characters. Furthermore, Kolors supports both Chinese and English inputs", "preview": "Kwai-Kolors.jpg", "skip": true, diff --git a/modules/model_kolors.py b/modules/model_kolors.py index a89af487b..932763b4b 100644 --- a/modules/model_kolors.py +++ b/modules/model_kolors.py @@ -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