diffusers/tools

1#!/usr/bin/env python32from diffusers import AutoPipelineForText2Image3import time4import os5from huggingface_hub import HfApi6import torch7from pathlib import Path8 9path = "stabilityai/stable-diffusion-xl-base-1.0"10 11api = HfApi()12start_time = time.time()13pipe = AutoPipelineForText2Image.from_pretrained(path, torch_dtype=torch.float16)14pipe.enable_model_cpu_offload()15lora_model_id = "hf-internal-testing/sdxl-0.9-kamepan-lora"16lora_model_id = "TheLastBen/Papercut_SDXL"17lora_filename = "kame_sdxl_v2-000020-16rank.safetensors"18lora_filename = "papercut.safetensors"19pipe.load_lora_weights(lora_model_id, weight_name=lora_filename)20 21prompt = "masterpiece, best quality, mountain"22prompt = "papercut sonic"23 24images = pipe(prompt=prompt, 25    num_inference_steps=20, 26    generator=torch.manual_seed(0)27).images28 29 30for i, image in enumerate(images):31    file_name = f"aa_{i}"32    path = os.path.join(Path.home(), "images", f"{file_name}.png")33    image.save(path)34 35    api.upload_file(36        path_or_fileobj=path,37        path_in_repo=path.split("/")[-1],38        repo_id="patrickvonplaten/images",39        repo_type="dataset",40    )41    print(f"https://huggingface.co/datasets/patrickvonplaten/images/blob/main/{file_name}.png")42