mirror of
https://github.com/kohya-ss/sd-scripts.git
synced 2026-04-09 06:45:09 +00:00
delet missed line
This commit is contained in:
@@ -259,7 +259,6 @@ def load_models_from_sdxl_checkpoint(model_version, ckpt_path, map_location, dty
|
|||||||
elif k.startswith("conditioner.embedders.1.model."):
|
elif k.startswith("conditioner.embedders.1.model."):
|
||||||
te2_sd[k] = state_dict.pop(k)
|
te2_sd[k] = state_dict.pop(k)
|
||||||
|
|
||||||
info1 = text_model1.load_state_dict(te1_sd)
|
|
||||||
info1 = _load_state_dict_on_device(text_model1, te1_sd, device=map_location) # remain fp32
|
info1 = _load_state_dict_on_device(text_model1, te1_sd, device=map_location) # remain fp32
|
||||||
print("text encoder 1:", info1)
|
print("text encoder 1:", info1)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user