Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -109,10 +109,12 @@ def error_str(error, title="Error"):
|
|
109 |
{error}""" if error else ""
|
110 |
|
111 |
def inference(prompt, guidance, steps, width=512, height=512, seed=0, img=None, strength=0.5, neg_prompt="", auto_prefix=False):
|
112 |
-
|
113 |
-
|
|
|
|
|
114 |
prompt = f"{prefix} {prompt}" if auto_prefix else prompt
|
115 |
-
|
116 |
try:
|
117 |
if img is not None:
|
118 |
return img_to_img(prompt, neg_prompt, img, strength, guidance, steps, width, height, generator), None
|
|
|
109 |
{error}""" if error else ""
|
110 |
|
111 |
def inference(prompt, guidance, steps, width=512, height=512, seed=0, img=None, strength=0.5, neg_prompt="", auto_prefix=False):
|
112 |
+
if torch.cuda.is_available():
|
113 |
+
generator = torch.Generator('cuda').manual_seed(seed) if seed != 0 else None
|
114 |
+
else:
|
115 |
+
generator = torch.Generator.manual_seed(seed) if seed != 0 else None
|
116 |
prompt = f"{prefix} {prompt}" if auto_prefix else prompt
|
117 |
+
|
118 |
try:
|
119 |
if img is not None:
|
120 |
return img_to_img(prompt, neg_prompt, img, strength, guidance, steps, width, height, generator), None
|