Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -15,11 +15,11 @@ def select_model(choice):#用于选择模型
|
|
15 |
pipe = None
|
16 |
paddle.device.cuda.empty_cache() #释放GPU显存
|
17 |
if choice == '文本引导图片风格变化':
|
18 |
-
pipe1 = StableDiffusionImg2ImgPipeline.from_pretrained("/
|
19 |
elif choice == '文本引导图像区域生成':
|
20 |
-
pipe1 = StableDiffusionInpaintPipeline.from_pretrained("/
|
21 |
elif choice == '文图引导图像区域生成':
|
22 |
-
pipe1 = PaintByExamplePipeline.from_pretrained("/
|
23 |
pipe = pipe1
|
24 |
return pipe1 #用于输出导入的模型相关信息
|
25 |
|
|
|
15 |
pipe = None
|
16 |
paddle.device.cuda.empty_cache() #释放GPU显存
|
17 |
if choice == '文本引导图片风格变化':
|
18 |
+
pipe1 = StableDiffusionImg2ImgPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
|
19 |
elif choice == '文本引导图像区域生成':
|
20 |
+
pipe1 = StableDiffusionInpaintPipeline.from_pretrained("stabilityai/stable-diffusion-2-inpainting")
|
21 |
elif choice == '文图引导图像区域生成':
|
22 |
+
pipe1 = PaintByExamplePipeline.from_pretrained("Fantasy-Studio/Paint-by-Example")
|
23 |
pipe = pipe1
|
24 |
return pipe1 #用于输出导入的模型相关信息
|
25 |
|