Spaces:
Paused
Paused
test gradio
Browse files
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
|
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 =
|
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 |
|