Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -11,7 +11,7 @@ import random
|
|
| 11 |
start_time = time.time()
|
| 12 |
current_steps = 15
|
| 13 |
|
| 14 |
-
pipe = DiffusionPipeline.from_pretrained("timbrooks/instruct-pix2pix", torch_dtype=torch.float16
|
| 15 |
pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
|
| 16 |
pipe.enable_xformers_memory_efficient_attention()
|
| 17 |
pipe.unet.to(memory_format=torch.channels_last)
|
|
|
|
| 11 |
start_time = time.time()
|
| 12 |
current_steps = 15
|
| 13 |
|
| 14 |
+
pipe = DiffusionPipeline.from_pretrained("timbrooks/instruct-pix2pix", torch_dtype=torch.float16, safety_checker=None)
|
| 15 |
pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
|
| 16 |
pipe.enable_xformers_memory_efficient_attention()
|
| 17 |
pipe.unet.to(memory_format=torch.channels_last)
|