JoPmt commited on
Commit
c7e949c
·
1 Parent(s): 6b944c6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.from_config_file("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")
 
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")