VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-08 22:35:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

add generation script

Browse Source
This commit is contained in:
ykume
2023-07-09 13:33:26 +09:00
parent fe7ede5af3
commit 1d25703ac3
2 changed files with 2548 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
library/sdxl_model_util.py
View File

@@ -182,7 +182,7 @@ def load_models_from_sdxl_checkpoint(model_version, ckpt_path, map_location):
converted_sd, logit_scale = convert_sdxl_text_encoder_2_checkpoint(te2_sd, max_length=77) converted_sd, logit_scale = convert_sdxl_text_encoder_2_checkpoint(te2_sd, max_length=77)
info2 = text_model2.load_state_dict(converted_sd) info2 = text_model2.load_state_dict(converted_sd)
print("text encoder2:", info2) print("text encoder 2:", info2)
# prepare vae # prepare vae
print("building VAE") print("building VAE")

2547
sdxl_gen_img.py Normal file
View File

File diff suppressed because it is too large Load Diff