r3gm commited on
Commit
953248b
·
verified ·
1 Parent(s): 66165d4

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +3 -1
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)