diff options
Diffstat (limited to 'infer.py')
-rw-r--r-- | infer.py | 2 |
1 files changed, 1 insertions, 1 deletions
@@ -32,7 +32,7 @@ from data.keywords import prompt_to_keywords, keywords_to_prompt | |||
32 | from models.clip.embeddings import patch_managed_embeddings | 32 | from models.clip.embeddings import patch_managed_embeddings |
33 | from models.clip.tokenizer import MultiCLIPTokenizer | 33 | from models.clip.tokenizer import MultiCLIPTokenizer |
34 | from pipelines.stable_diffusion.vlpn_stable_diffusion import VlpnStableDiffusion | 34 | from pipelines.stable_diffusion.vlpn_stable_diffusion import VlpnStableDiffusion |
35 | from util import load_config, load_embeddings_from_dir | 35 | from util.files import load_config, load_embeddings_from_dir |
36 | 36 | ||
37 | 37 | ||
38 | torch.backends.cuda.matmul.allow_tf32 = True | 38 | torch.backends.cuda.matmul.allow_tf32 = True |