diffusers/tools

1#!/usr/bin/env python32#!/usr/bin/env python33from diffusers import DiffusionPipeline4import torch5import time6import os7from pathlib import Path8from huggingface_hub import HfApi9 10api = HfApi()11start_time = time.time()12 13pipe = DiffusionPipeline.from_pretrained("/home/patrick/if", torch_dtype=torch.float16, variant="fp16", use_safetensors=True)14pipe.enable_model_cpu_offload()15 16generator = torch.Generator("cuda").manual_seed(0)17prompt = 'a photo of a kangaroo wearing an orange hoodie and blue sunglasses standing in front of the eiffel tower holding a sign that says "very deep learning"'18 19image = pipe(prompt, generator=generator).images[0]20 21path = os.path.join(Path.home(), "images", "if.png")22image.save(path)23 24api.upload_file(25    path_or_fileobj=path,26    path_in_repo=path.split("/")[-1],27    repo_id="patrickvonplaten/images",28    repo_type="dataset",29)30print(f"https://huggingface.co/datasets/patrickvonplaten/images/blob/main/if.png")31