Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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=[]
|