Spaces:
Runtime error
Runtime error
Commit
·
e358234
1
Parent(s):
71147d5
Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ pipe = DiffusionPipeline.from_pretrained("timbrooks/instruct-pix2pix", torch_dty
|
|
16 |
pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
|
17 |
pipe.enable_xformers_memory_efficient_attention()
|
18 |
pipe.unet.to(memory_format=torch.channels_last)
|
19 |
-
pipe = pipe.to("
|
20 |
|
21 |
s3_access_key = "juj22qxqxql7u2pl6nomgxu3ip7a"
|
22 |
s3_secret_key = "j3uwidtozhboy5vczhymhzkkjsaumznnqlzck5zjs5qxgsung4ukk"
|
|
|
16 |
pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
|
17 |
pipe.enable_xformers_memory_efficient_attention()
|
18 |
pipe.unet.to(memory_format=torch.channels_last)
|
19 |
+
pipe = pipe.to("CPU")
|
20 |
|
21 |
s3_access_key = "juj22qxqxql7u2pl6nomgxu3ip7a"
|
22 |
s3_secret_key = "j3uwidtozhboy5vczhymhzkkjsaumznnqlzck5zjs5qxgsung4ukk"
|