Hugging Face's logo

Spaces:
xywwww
/
scene_diffusion___
Runtime error

App Files Community
1
xywwww commited on
Commit
d42ce16
·
verified ·
1 Parent(s): 370c493

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -70,7 +70,7 @@ import numpy as np
70
 
71
  # 初始化 ControlNet 模型和 Stable Diffusion Pipeline
72
  controlnet = ControlNetModel.from_pretrained("lllyasviel/sd-controlnet-canny", torch_dtype=torch.float16)
73
- pipeline = StableDiffusionControlNetPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", controlnet=controlnet).to("cuda")
74
 
75
  # 定义图像生成函数
76
  def generate_image(prompt: str, input_image: Image.Image):
 
70
 
71
  # 初始化 ControlNet 模型和 Stable Diffusion Pipeline
72
  controlnet = ControlNetModel.from_pretrained("lllyasviel/sd-controlnet-canny", torch_dtype=torch.float16)
73
+ pipeline = StableDiffusionControlNetPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", controlnet=controlnet)
74
 
75
  # 定义图像生成函数
76
  def generate_image(prompt: str, input_image: Image.Image):