diff options
Diffstat (limited to 'infer.py')
-rw-r--r-- | infer.py | 4 |
1 files changed, 2 insertions, 2 deletions
@@ -245,8 +245,8 @@ def create_pipeline(model, dtype): | |||
245 | tokenizer=tokenizer, | 245 | tokenizer=tokenizer, |
246 | scheduler=scheduler, | 246 | scheduler=scheduler, |
247 | ) | 247 | ) |
248 | # pipeline.enable_xformers_memory_efficient_attention() | 248 | pipeline.enable_xformers_memory_efficient_attention() |
249 | pipeline.unet = torch.compile(pipeline.unet) | 249 | # pipeline.unet = torch.compile(pipeline.unet) |
250 | pipeline.enable_vae_slicing() | 250 | pipeline.enable_vae_slicing() |
251 | pipeline.to("cuda") | 251 | pipeline.to("cuda") |
252 | 252 | ||