add generation script

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

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)
info2 = text_model2.load_state_dict(converted_sd)
print("text encoder2:", info2)
print("text encoder 2:", info2)
# prepare vae
print("building VAE")

2547
sdxl_gen_img.py Normal file

File diff suppressed because it is too large Load Diff