sdxl create_sdxl_fsdp_configs.py data_process.py extract_lora_module.py generate_noise_image_pairs_laion_sdxl.py generate_vae_latents.py sdxl_ode_dataset.py sdxl_text_encoder.py test_folder_sdxl.py sd_guidance.py sd_image_dataset.py ...
4-step LoRA generation importtorchfromdiffusersimportDiffusionPipeline,UNet2DConditionModel,LCMSchedulerfromhuggingface_hubimporthf_hub_downloadfromsafetensors.torchimportload_filebase_model_id="stabilityai/stable-diffusion-xl-base-1.0"repo_name="tianweiy/DMD2"ckpt_name="dmd2_sdxl_4step_lora_fp16.safe...
torch import load_file base_model_id = "stabilityai/stable-diffusion-xl-base-1.0" repo_name = "tianweiy/DMD2" ckpt_name = "dmd2_sdxl_4step_lora_fp16.safetensors" # Load model. pipe = DiffusionPipeline.from_pretrained(base_model_id, torch_dtype=torch.float16, variant="fp16").to(...