amos1088 commited on
Commit
1ac2f7d
·
1 Parent(s): 9b01735

test gradio

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
  import torch
3
  # from diffusers import StableDiffusion3ControlNetPipeline, SD3ControlNetModel, UniPCMultistepScheduler
4
- from diffusers import AutoPipelineForText2Image
5
  from huggingface_hub import login
6
  import os
7
  import spaces
@@ -27,7 +27,7 @@ login(token=token)
27
  # controlnet = SD3ControlNetModel.from_pretrained("alimama-creative/SD3-Controlnet-Softedge", torch_dtype=torch.float16)
28
 
29
  # pipe = StableDiffusion3ControlNetPipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", controlnet=controlnet)
30
- pipe = AutoPipelineForText2Image.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers")
31
  pipe.load_ip_adapter("TencentARC/T2I-Adapter", subfolder="models", weight_name="coadapter-style-sd15v1.pth")
32
  pipe.to("cuda", torch.float16)
33
 
 
1
  import gradio as gr
2
  import torch
3
  # from diffusers import StableDiffusion3ControlNetPipeline, SD3ControlNetModel, UniPCMultistepScheduler
4
+ from diffusers import StableDiffusionXLPipeline
5
  from huggingface_hub import login
6
  import os
7
  import spaces
 
27
  # controlnet = SD3ControlNetModel.from_pretrained("alimama-creative/SD3-Controlnet-Softedge", torch_dtype=torch.float16)
28
 
29
  # pipe = StableDiffusion3ControlNetPipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", controlnet=controlnet)
30
+ pipe = StableDiffusionXLPipeline.from_pretrained("stabilityai/sdxl-turbo")
31
  pipe.load_ip_adapter("TencentARC/T2I-Adapter", subfolder="models", weight_name="coadapter-style-sd15v1.pth")
32
  pipe.to("cuda", torch.float16)
33