Skip to content

Commit b2a77da

Browse files
committed
Fix long prompts crashing the app
1 parent 6ddac71 commit b2a77da

File tree

2 files changed

+10
-2
lines changed

2 files changed

+10
-2
lines changed

README.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
All in one, batteries-included software to easily generate and upscale AI art using [Stable Diffusion](https://github.com/CompVis/stable-diffusion).
44

55
![Image](https://i.imgur.com/EyjdVXj.png)
6+
![Image](https://i.imgur.com/kB3mL6e.png)
67

78
## Requirements / Info
89

SDToolkit/Generator.cs

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,13 @@ public static void GenerateFromPrompt(GeneratorConfig config)
6262

6363
private static void Run(GeneratorConfig config)
6464
{
65-
Directory.Delete(WorkingDirectory + "\\stable-diffusion\\outputs\\txt2img-samples", true);
65+
var samplesDir = WorkingDirectory + "\\stable-diffusion\\outputs\\txt2img-samples";
66+
if (Directory.Exists(samplesDir))
67+
{
68+
Directory.Delete(samplesDir, true);
69+
}
70+
71+
var path = DateTimeOffset.Now.ToUnixTimeSeconds().ToString();
6672

6773
config.GenerateButton.Invoke(new MethodInvoker(delegate ()
6874
{
@@ -89,7 +95,8 @@ private static void Run(GeneratorConfig config)
8995
var prompt = config.Prompt;
9096
var converted = string.Join("_", prompt.Split(' '));
9197

92-
var images = Directory.GetFiles(WorkingDirectory + @"\stable-diffusion\outputs\txt2img-samples\" + converted, "*.png");
98+
var dir = Directory.GetDirectories(WorkingDirectory + @"\stable-diffusion\outputs\txt2img-samples\")[0];
99+
var images = Directory.GetFiles(dir, "*.png");
93100

94101
config.GenerateButton.Invoke(new MethodInvoker(delegate ()
95102
{

0 commit comments

Comments
 (0)