Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,7 @@ from diffusers import EulerDiscreteScheduler, UNet2DConditionModel, AutoencoderK
|
|
11 |
from gradio_client import Client
|
12 |
|
13 |
accelerator = Accelerator(cpu=True)
|
14 |
-
pipe = accelerator.prepare(DiffusionPipeline.
|
15 |
pipe.scheduler = EulerDiscreteScheduler.from_config("stabilityai/sd-turbo", subfolder="scheduler")
|
16 |
pipe.text_encoder = CLIPTextModel.from_config("stabilityai/sd-turbo", subfolder="text_encoder")
|
17 |
pipe.tokenizer = CLIPTokenizer.from_config("stabilityai/sd-turbo", subfolder="tokenizer")
|
|
|
11 |
from gradio_client import Client
|
12 |
|
13 |
accelerator = Accelerator(cpu=True)
|
14 |
+
pipe = accelerator.prepare(DiffusionPipeline.from_config("stabilityai/sd-turbo", torch_dtype=torch.float32, use_safetensors=True, safety_checker=None))
|
15 |
pipe.scheduler = EulerDiscreteScheduler.from_config("stabilityai/sd-turbo", subfolder="scheduler")
|
16 |
pipe.text_encoder = CLIPTextModel.from_config("stabilityai/sd-turbo", subfolder="text_encoder")
|
17 |
pipe.tokenizer = CLIPTokenizer.from_config("stabilityai/sd-turbo", subfolder="tokenizer")
|