1 files changed, 1 insertions, 1 deletions
diff --git a/infer.py b/infer.py
index 5bd926a..f607041 100644
--- a/infer.py
+++ b/infer.py
@@ -31,7 +31,7 @@ torch.backends.cudnn.benchmark = True
 default_args = {
-    "model": None,
+    "model": "stabilityai/stable-diffusion-2-1",
    "precision": "fp32",
    "ti_embeddings_dir": "embeddings_ti",
    "output_dir": "output/inference",

diff --git a/infer.py b/infer.py index 5bd926a..f607041 100644 --- a/infer.py +++ b/infer.py
@@ -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",