from diffusers import StableDiffusionXLPipeline pipe = StableDiffusionXLPipeline.from_single_file("/root/Dreambooth/experiments/chair1/0/unet/diffusion_pytorch_model.safetensors") pipe.save_pretrained("./")