Merge branch 'main' of https://huggingface.co/diffusers/tools into main

run_local.py

@@ -2,14 +2,18 @@
 from diffusers import StableDiffusionPipeline, DPMSolverSinglestepScheduler, DPMSolverMultistepScheduler, DEISMultistepScheduler, HeunDiscreteScheduler
 from diffusers.schedulers.scheduling_unipc_multistep import UniPCMultistepScheduler
 import time
+import os
 from huggingface_hub import HfApi
 import torch
 import sys
+from pathlib import Path
 
 path = sys.argv[1]
 
 api = HfApi()
 start_time = time.time()
+#pipe = StableDiffusionPipeline.from_pretrained(path, torch_dtype=torch.float16, device_map="auto")
+#pipe.scheduler = HeunDiscreteScheduler.from_config(pipe.scheduler.config)
 pipe = StableDiffusionPipeline.from_pretrained(path, torch_dtype=torch.float16)
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
 pipe = pipe.to("cuda")
@@ -20,7 +24,7 @@ generator = torch.Generator(device="cuda").manual_seed(0)
 image = pipe(prompt, generator=generator, num_inference_steps=15).images[0]
 print("Time", time.time() - start_time)
 
-path = "/home/patrick_huggingface_co/images/aa.png"
+path = os.path.join(Path.home(), "images", "aa.png")
 image.save(path)
 
 api.upload_file(