Hugging Face's logo

Spaces:
chaowei100
/
ChatGPT_Taiyi-Stable-Diffusion
Runtime error

App Files Community
chaowei100 commited on
Commit
6227d6e
·
1 Parent(s): 6462634

Update funtional_picture.py

Browse files
Files changed (1) hide show
  1. funtional_picture.py +1 -1
funtional_picture.py CHANGED
@@ -8,7 +8,7 @@ from diffusers import (
8
  StableDiffusionInpaintPipeline,
9
  )
10
  device="cuda"
11
- model_id = "./Taiyi-Stable-Diffusion-1B-Chinese-v0.1"
12
 
13
  pipe_text2img = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16).to(device)
14
  pipe_img2img = StableDiffusionImg2ImgPipeline(**pipe_text2img.components).to(device)
 
8
  StableDiffusionInpaintPipeline,
9
  )
10
  device="cuda"
11
+ model_id = "IDEA-CCNL/Taiyi-Stable-Diffusion-1B-Chinese-v0.1"
12
 
13
  pipe_text2img = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16).to(device)
14
  pipe_img2img = StableDiffusionImg2ImgPipeline(**pipe_text2img.components).to(device)