diff options
Diffstat (limited to 'infer.py')
-rw-r--r-- | infer.py | 2 |
1 files changed, 1 insertions, 1 deletions
@@ -31,7 +31,7 @@ torch.backends.cudnn.benchmark = True | |||
31 | 31 | ||
32 | 32 | ||
33 | default_args = { | 33 | default_args = { |
34 | "model": None, | 34 | "model": "stabilityai/stable-diffusion-2-1", |
35 | "precision": "fp32", | 35 | "precision": "fp32", |
36 | "ti_embeddings_dir": "embeddings_ti", | 36 | "ti_embeddings_dir": "embeddings_ti", |
37 | "output_dir": "output/inference", | 37 | "output_dir": "output/inference", |