diff options
| author | Volpeon <git@volpeon.ink> | 2023-02-17 15:56:54 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-02-17 15:56:54 +0100 |
| commit | c927eeb3516b8ecae201441a68956f32247a6d7c (patch) | |
| tree | b24e4246a067415151039484d1f8da2658687ef3 /infer.py | |
| parent | Remove xformers, switch to Pytorch Nightly (diff) | |
| download | textual-inversion-diff-c927eeb3516b8ecae201441a68956f32247a6d7c.tar.gz textual-inversion-diff-c927eeb3516b8ecae201441a68956f32247a6d7c.tar.bz2 textual-inversion-diff-c927eeb3516b8ecae201441a68956f32247a6d7c.zip | |
Back to xformers
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 | ||