Spaces:

turboedit
/

turbo_edit

Running on Zero

turboedit commited on Aug 1

Commit

3347638

•

1 Parent(s): 128983c

Upload my_run.py with huggingface_hub

Files changed (1) hide show

my_run.py CHANGED Viewed

@@ -331,7 +331,7 @@ class myDDPMScheduler(DDPMScheduler):
 pipeline = AutoPipelineForImage2Image.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16", safety_checker = None)
-pipeline = pipeline.to("cuda")
 pipeline.scheduler = DDPMScheduler.from_pretrained(  # type: ignore
                 'stabilityai/sdxl-turbo',
                 subfolder="scheduler",

 pipeline = AutoPipelineForImage2Image.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16", safety_checker = None)
+pipeline = pipeline.to(device)
 pipeline.scheduler = DDPMScheduler.from_pretrained(  # type: ignore
                 'stabilityai/sdxl-turbo',
                 subfolder="scheduler",