add historical config feature
Browse files
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"
|