Update app.py

app.py

@@ -11,10 +11,10 @@ from diffusers import EulerDiscreteScheduler, UNet2DConditionModel, AutoencoderK
 from gradio_client import Client
 
 accelerator = Accelerator(cpu=True)
-pipe = accelerator.prepare(DiffusionPipeline.from_config("stabilityai/sd-turbo", torch_dtype=torch.float32, use_safetensors=True, safety_checker=None))
-pipe.scheduler = EulerDiscreteScheduler.from_config("stabilityai/sd-turbo", subfolder="scheduler")
-pipe.text_encoder = CLIPTextModel.from_config("stabilityai/sd-turbo", subfolder="text_encoder")
-pipe.tokenizer = CLIPTokenizer.from_config("stabilityai/sd-turbo", subfolder="tokenizer")
+pipe = accelerator.prepare(DiffusionPipeline.from_pretrained("stabilityai/sd-turbo", torch_dtype=torch.float32, use_safetensors=True, safety_checker=None))
+pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
+pipe.text_encoder = CLIPTextModel._from_config(pipe.text_encoder.config)
+pipe.tokenizer = CLIPTokenizer.from_config(pipe.tokenizer.config)
 pipe.UNet2DConditionModel = UNet2DConditionModel.from_config("stabilityai/sd-turbo", subfolder="unet")
 pipe.AutoencoderKL = AutoencoderKL.from_config("stabilityai/sd-turbo", subfolder="vae")
 pipe = accelerator.prepare(pipe.to("cpu"))