diff options
Diffstat (limited to 'infer.py')
-rw-r--r-- | infer.py | 2 |
1 files changed, 1 insertions, 1 deletions
@@ -234,7 +234,7 @@ def create_pipeline(model, scheduler, ti_embeddings_dir, dtype): | |||
234 | tokenizer=tokenizer, | 234 | tokenizer=tokenizer, |
235 | scheduler=scheduler, | 235 | scheduler=scheduler, |
236 | ) | 236 | ) |
237 | pipeline.aesthetic_gradient_iters = 20 | 237 | pipeline.enable_xformers_memory_efficient_attention() |
238 | pipeline.to("cuda") | 238 | pipeline.to("cuda") |
239 | 239 | ||
240 | print("Pipeline loaded.") | 240 | print("Pipeline loaded.") |