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