Update funtional_picture.py

funtional_picture.py

@@ -10,8 +10,8 @@ from diffusers import (
 device="cuda"
 model_id = "IDEA-CCNL/Taiyi-Stable-Diffusion-1B-Chinese-v0.1"
 
-pipe_text2img = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16).to(device)
-pipe_img2img = StableDiffusionImg2ImgPipeline(**pipe_text2img.components).to(device)
+pipe_text2img = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+pipe_img2img = StableDiffusionImg2ImgPipeline(**pipe_text2img.components)
 def infer_text2img(prompt, guide, steps, width, height, image_in, strength):
     if image_in is not None:
         init_image = image_in.convert("RGB").resize((width, height))