amos1088 commited on
Commit
91e7ec6
·
1 Parent(s): 505f3d2

test gradio

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import gradio as gr
2
  import torch
3
- from diffusers import StableDiffusion3ControlNetPipeline, ControlNetModel, UniPCMultistepScheduler
4
  from huggingface_hub import login
5
  import os
6
  import spaces
@@ -10,10 +10,10 @@ login(token=token)
10
 
11
  # Model IDs for Stable Diffusion 1.5 and ControlNet
12
  model_id = "stabilityai/stable-diffusion-3.5-large-turbo"
13
- controlnet_id = "lllyasviel/control_v11p_sd15_inpaint"
14
 
15
  # Load the ControlNet model and Stable Diffusion pipeline
16
- controlnet = ControlNetModel.from_pretrained(controlnet_id, torch_dtype=torch.float16)
17
  pipe = StableDiffusion3ControlNetPipeline.from_pretrained(
18
  model_id, controlnet=controlnet, torch_dtype=torch.float16
19
  )
 
1
  import gradio as gr
2
  import torch
3
+ from diffusers import StableDiffusion3ControlNetPipeline, SD3ControlNetModel, UniPCMultistepScheduler
4
  from huggingface_hub import login
5
  import os
6
  import spaces
 
10
 
11
  # Model IDs for Stable Diffusion 1.5 and ControlNet
12
  model_id = "stabilityai/stable-diffusion-3.5-large-turbo"
13
+ controlnet_id = "InstantX/SD3-Controlnet-Tile"
14
 
15
  # Load the ControlNet model and Stable Diffusion pipeline
16
+ controlnet = SD3ControlNetModel.from_pretrained(controlnet_id, torch_dtype=torch.float16)
17
  pipe = StableDiffusion3ControlNetPipeline.from_pretrained(
18
  model_id, controlnet=controlnet, torch_dtype=torch.float16
19
  )