groggy84 commited on
Commit
1edfe5e
1 Parent(s): 79118df

cuda disable

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ neg_prompt = "chinese letter"
13
  controlnet_repo_id = "calihyper/trad-kor-controlnet"
14
  repo_id = "calihyper/trad-kor-landscape-black"
15
  controlnet = ControlNetModel.from_pretrained(controlnet_repo_id)
16
- pipe = StableDiffusionControlNetPipeline.from_pretrained(repo_id, controlnet=controlnet).to("cuda")
17
 
18
 
19
  def change_radio(input):
 
13
  controlnet_repo_id = "calihyper/trad-kor-controlnet"
14
  repo_id = "calihyper/trad-kor-landscape-black"
15
  controlnet = ControlNetModel.from_pretrained(controlnet_repo_id)
16
+ pipe = StableDiffusionControlNetPipeline.from_pretrained(repo_id, controlnet=controlnet)
17
 
18
 
19
  def change_radio(input):