JoPmt commited on
Commit
5a1bdb7
·
1 Parent(s): 9a59bfc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -14,9 +14,9 @@ accelerator = Accelerator(cpu=True)
14
  pipe = accelerator.prepare(DiffusionPipeline.from_pretrained("stabilityai/sd-turbo", torch_dtype=torch.float32, use_safetensors=True, safety_checker=None))
15
  pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
16
  pipe.text_encoder = CLIPTextModel._from_config(pipe.text_encoder.config)
17
- pipe.tokenizer = CLIPTokenizer.from_config(pipe.tokenizer.config)
18
- pipe.UNet2DConditionModel = UNet2DConditionModel.from_config("stabilityai/sd-turbo", subfolder="unet")
19
- pipe.AutoencoderKL = AutoencoderKL.from_config("stabilityai/sd-turbo", subfolder="vae")
20
  pipe = accelerator.prepare(pipe.to("cpu"))
21
  generator = torch.Generator("cpu").manual_seed(random.randint(1, 867346))
22
  apol=[]
 
14
  pipe = accelerator.prepare(DiffusionPipeline.from_pretrained("stabilityai/sd-turbo", torch_dtype=torch.float32, use_safetensors=True, safety_checker=None))
15
  pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
16
  pipe.text_encoder = CLIPTextModel._from_config(pipe.text_encoder.config)
17
+ ##pipe.tokenizer = CLIPTokenizer.from_config(pipe.tokenizer.config)
18
+ ##pipe.UNet2DConditionModel = UNet2DConditionModel.from_config("stabilityai/sd-turbo", subfolder="unet")
19
+ ##pipe.AutoencoderKL = AutoencoderKL.from_config("stabilityai/sd-turbo", subfolder="vae")
20
  pipe = accelerator.prepare(pipe.to("cpu"))
21
  generator = torch.Generator("cpu").manual_seed(random.randint(1, 867346))
22
  apol=[]