Spaces:
Running on Zero
Running on Zero
Update utils.py
Browse files
utils.py
CHANGED
|
@@ -475,7 +475,6 @@ def get_model_list(directory_path):
|
|
| 475 |
|
| 476 |
def extract_parameters(input_string):
|
| 477 |
parameters = {}
|
| 478 |
-
input_string = input_string.replace("\n", "")
|
| 479 |
|
| 480 |
if "Negative prompt:" not in input_string:
|
| 481 |
if "Steps:" in input_string:
|
|
@@ -489,12 +488,15 @@ def extract_parameters(input_string):
|
|
| 489 |
|
| 490 |
parm = input_string.split("Negative prompt:")
|
| 491 |
parameters["prompt"] = parm[0].strip()
|
|
|
|
| 492 |
if "Steps:" not in parm[1]:
|
| 493 |
parameters["neg_prompt"] = parm[1].strip()
|
| 494 |
return parameters
|
|
|
|
| 495 |
parm = parm[1].split("Steps:")
|
| 496 |
parameters["neg_prompt"] = parm[0].strip()
|
| 497 |
input_string = "Steps:" + parm[1]
|
|
|
|
| 498 |
|
| 499 |
# Extracting Steps
|
| 500 |
steps_match = re.search(r'Steps: (\d+)', input_string)
|
|
|
|
| 475 |
|
| 476 |
def extract_parameters(input_string):
|
| 477 |
parameters = {}
|
|
|
|
| 478 |
|
| 479 |
if "Negative prompt:" not in input_string:
|
| 480 |
if "Steps:" in input_string:
|
|
|
|
| 488 |
|
| 489 |
parm = input_string.split("Negative prompt:")
|
| 490 |
parameters["prompt"] = parm[0].strip()
|
| 491 |
+
|
| 492 |
if "Steps:" not in parm[1]:
|
| 493 |
parameters["neg_prompt"] = parm[1].strip()
|
| 494 |
return parameters
|
| 495 |
+
|
| 496 |
parm = parm[1].split("Steps:")
|
| 497 |
parameters["neg_prompt"] = parm[0].strip()
|
| 498 |
input_string = "Steps:" + parm[1]
|
| 499 |
+
input_string = input_string.replace("\n", "").strip()
|
| 500 |
|
| 501 |
# Extracting Steps
|
| 502 |
steps_match = re.search(r'Steps: (\d+)', input_string)
|