n42 commited on
Commit
d680fb0
·
1 Parent(s): e0341a3

add historical config feature

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -126,6 +126,8 @@ def inference_steps_change(inference_steps):
126
 
127
  def in_manual_seed_change(manual_seed):
128
 
 
 
129
  if manual_seed < 0 or manual_seed is None or manual_seed == '':
130
  code[code_pos_manual_seed] = f'# manual_seed = {manual_seed}'
131
  generator = f'generator = torch.Generator("{device}")'
@@ -279,8 +281,9 @@ def run_inference(model,
279
  generator=generator,
280
  num_inference_steps=int(inference_steps),
281
  guidance_scale=float(guidance_scale)).images[0]
282
-
283
- config_history.append(config)
 
284
  return image, dict_list_to_markdown_table(config_history)
285
 
286
  else:
@@ -296,7 +299,10 @@ def dict_list_to_markdown_table(config_history):
296
  markdown_table = "| share | " + " | ".join(headers) + " |\n"
297
  markdown_table += "| --- | " + " | ".join(["---"] * len(headers)) + " |\n"
298
 
 
 
299
  for index, config in enumerate(config_history):
 
300
  encoded_config = base64.b64encode(str(config).encode()).decode()
301
  share_link = f'<a target="_blank" href="?config={encoded_config}">📎</a>'
302
  markdown_table += f"| {share_link} | " + " | ".join(str(config.get(key, "")) for key in headers) + " |\n"
 
126
 
127
  def in_manual_seed_change(manual_seed):
128
 
129
+ config['manual_seed'] = manual_seed
130
+
131
  if manual_seed < 0 or manual_seed is None or manual_seed == '':
132
  code[code_pos_manual_seed] = f'# manual_seed = {manual_seed}'
133
  generator = f'generator = torch.Generator("{device}")'
 
281
  generator=generator,
282
  num_inference_steps=int(inference_steps),
283
  guidance_scale=float(guidance_scale)).images[0]
284
+
285
+ config_history.append(config.copy())
286
+
287
  return image, dict_list_to_markdown_table(config_history)
288
 
289
  else:
 
299
  markdown_table = "| share | " + " | ".join(headers) + " |\n"
300
  markdown_table += "| --- | " + " | ".join(["---"] * len(headers)) + " |\n"
301
 
302
+ print('#######################')
303
+
304
  for index, config in enumerate(config_history):
305
+
306
  encoded_config = base64.b64encode(str(config).encode()).decode()
307
  share_link = f'<a target="_blank" href="?config={encoded_config}">📎</a>'
308
  markdown_table += f"| {share_link} | " + " | ".join(str(config.get(key, "")) for key in headers) + " |\n"