diff options
| author | Volpeon <git@volpeon.ink> | 2022-10-09 12:42:21 +0200 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2022-10-09 12:42:21 +0200 |
| commit | 1eef9a946161fd06b0e72ec804c68f4f0e74b380 (patch) | |
| tree | b4a272b7240c25c0eef173dbfd193ba89a592929 /infer.py | |
| parent | Update (diff) | |
| download | textual-inversion-diff-1eef9a946161fd06b0e72ec804c68f4f0e74b380.tar.gz textual-inversion-diff-1eef9a946161fd06b0e72ec804c68f4f0e74b380.tar.bz2 textual-inversion-diff-1eef9a946161fd06b0e72ec804c68f4f0e74b380.zip | |
Update
Diffstat (limited to 'infer.py')
| -rw-r--r-- | infer.py | 4 |
1 files changed, 2 insertions, 2 deletions
| @@ -11,8 +11,9 @@ from PIL import Image | |||
| 11 | from diffusers import AutoencoderKL, UNet2DConditionModel, PNDMScheduler, DDIMScheduler, LMSDiscreteScheduler | 11 | from diffusers import AutoencoderKL, UNet2DConditionModel, PNDMScheduler, DDIMScheduler, LMSDiscreteScheduler |
| 12 | from transformers import CLIPTextModel, CLIPTokenizer | 12 | from transformers import CLIPTextModel, CLIPTokenizer |
| 13 | from slugify import slugify | 13 | from slugify import slugify |
| 14 | from pipelines.stable_diffusion.vlpn_stable_diffusion import VlpnStableDiffusion | 14 | |
| 15 | from schedulers.scheduling_euler_a import EulerAScheduler | 15 | from schedulers.scheduling_euler_a import EulerAScheduler |
| 16 | from pipelines.stable_diffusion.vlpn_stable_diffusion import VlpnStableDiffusion | ||
| 16 | 17 | ||
| 17 | 18 | ||
| 18 | torch.backends.cuda.matmul.allow_tf32 = True | 19 | torch.backends.cuda.matmul.allow_tf32 = True |
| @@ -235,7 +236,6 @@ def create_pipeline(model, scheduler, embeddings_dir, dtype): | |||
| 235 | tokenizer=tokenizer, | 236 | tokenizer=tokenizer, |
| 236 | scheduler=scheduler, | 237 | scheduler=scheduler, |
| 237 | ) | 238 | ) |
| 238 | # pipeline.enable_attention_slicing() | ||
| 239 | pipeline.to("cuda") | 239 | pipeline.to("cuda") |
| 240 | 240 | ||
| 241 | print("Pipeline loaded.") | 241 | print("Pipeline loaded.") |