diff options
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.") |