Spaces:

sdart
/

SD_txt2img

Runtime error

sdart commited on May 16, 2023

Commit

1e0e2d1

1 Parent(s): 1ecf68a

update steps

Files changed (1) hide show

model.py CHANGED Viewed

@@ -19,7 +19,7 @@ class Model:
                 negative_prompt: str,
                 guidance_scale:int = 7,
                 num_images:int = 1,
-                num_steps:int = 2,
                 ):
         seed = np.random.randint(0, np.iinfo(np.int64).max)
         generator = torch.Generator().manual_seed(seed)

                 negative_prompt: str,
                 guidance_scale:int = 7,
                 num_images:int = 1,
+                num_steps:int = 20,
                 ):
         seed = np.random.randint(0, np.iinfo(np.int64).max)
         generator = torch.Generator().manual_seed(seed)