Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -103,6 +103,8 @@ def generate_images_30(prompt, neg_prompt_1, neg_prompt_2, neg_prompt_3, width,
|
|
| 103 |
seed = random.randint(0, MAX_SEED)
|
| 104 |
generator = torch.Generator(device=device).manual_seed(seed)
|
| 105 |
print('-- generating image --')
|
|
|
|
|
|
|
| 106 |
sd_image = pipe(
|
| 107 |
prompt=prompt, prompt_2=prompt, prompt_3=prompt,
|
| 108 |
negative_prompt=neg_prompt_1, negative_prompt_2=neg_prompt_2, negative_prompt_3=neg_prompt_3,
|
|
@@ -111,6 +113,8 @@ def generate_images_30(prompt, neg_prompt_1, neg_prompt_2, neg_prompt_3, width,
|
|
| 111 |
max_sequence_length=384
|
| 112 |
).images[0]
|
| 113 |
print('-- got image --')
|
|
|
|
|
|
|
| 114 |
with torch.no_grad():
|
| 115 |
upscale = upscaler_2(sd_image, tiling=True, tile_width=256, tile_height=256)
|
| 116 |
upscale2 = upscaler_2(upscale, tiling=True, tile_width=256, tile_height=256)
|
|
@@ -123,6 +127,8 @@ def generate_images_60(prompt, neg_prompt_1, neg_prompt_2, neg_prompt_3, width,
|
|
| 123 |
seed = random.randint(0, MAX_SEED)
|
| 124 |
generator = torch.Generator(device=device).manual_seed(seed)
|
| 125 |
print('-- generating image --')
|
|
|
|
|
|
|
| 126 |
sd_image = pipe(
|
| 127 |
prompt=prompt, prompt_2=prompt, prompt_3=prompt,
|
| 128 |
negative_prompt=neg_prompt_1, negative_prompt_2=neg_prompt_2, negative_prompt_3=neg_prompt_3,
|
|
@@ -131,6 +137,8 @@ def generate_images_60(prompt, neg_prompt_1, neg_prompt_2, neg_prompt_3, width,
|
|
| 131 |
max_sequence_length=384
|
| 132 |
).images[0]
|
| 133 |
print('-- got image --')
|
|
|
|
|
|
|
| 134 |
with torch.no_grad():
|
| 135 |
upscale = upscaler_2(sd_image, tiling=True, tile_width=256, tile_height=256)
|
| 136 |
upscale2 = upscaler_2(upscale, tiling=True, tile_width=256, tile_height=256)
|
|
@@ -143,6 +151,8 @@ def generate_images_110(prompt, neg_prompt_1, neg_prompt_2, neg_prompt_3, width,
|
|
| 143 |
seed = random.randint(0, MAX_SEED)
|
| 144 |
generator = torch.Generator(device=device).manual_seed(seed)
|
| 145 |
print('-- generating image --')
|
|
|
|
|
|
|
| 146 |
sd_image = pipe(
|
| 147 |
prompt=prompt, prompt_2=prompt, prompt_3=prompt,
|
| 148 |
negative_prompt=neg_prompt_1, negative_prompt_2=neg_prompt_2, negative_prompt_3=neg_prompt_3,
|
|
@@ -151,6 +161,8 @@ def generate_images_110(prompt, neg_prompt_1, neg_prompt_2, neg_prompt_3, width,
|
|
| 151 |
max_sequence_length=384
|
| 152 |
).images[0]
|
| 153 |
print('-- got image --')
|
|
|
|
|
|
|
| 154 |
with torch.no_grad():
|
| 155 |
upscale = upscaler_2(sd_image, tiling=True, tile_width=256, tile_height=256)
|
| 156 |
upscale2 = upscaler_2(upscale, tiling=True, tile_width=256, tile_height=256)
|
|
|
|
| 103 |
seed = random.randint(0, MAX_SEED)
|
| 104 |
generator = torch.Generator(device=device).manual_seed(seed)
|
| 105 |
print('-- generating image --')
|
| 106 |
+
torch.cuda.empty_cache()
|
| 107 |
+
torch.cuda.reset_peak_memory_stats()
|
| 108 |
sd_image = pipe(
|
| 109 |
prompt=prompt, prompt_2=prompt, prompt_3=prompt,
|
| 110 |
negative_prompt=neg_prompt_1, negative_prompt_2=neg_prompt_2, negative_prompt_3=neg_prompt_3,
|
|
|
|
| 113 |
max_sequence_length=384
|
| 114 |
).images[0]
|
| 115 |
print('-- got image --')
|
| 116 |
+
torch.cuda.empty_cache()
|
| 117 |
+
torch.cuda.reset_peak_memory_stats()
|
| 118 |
with torch.no_grad():
|
| 119 |
upscale = upscaler_2(sd_image, tiling=True, tile_width=256, tile_height=256)
|
| 120 |
upscale2 = upscaler_2(upscale, tiling=True, tile_width=256, tile_height=256)
|
|
|
|
| 127 |
seed = random.randint(0, MAX_SEED)
|
| 128 |
generator = torch.Generator(device=device).manual_seed(seed)
|
| 129 |
print('-- generating image --')
|
| 130 |
+
torch.cuda.empty_cache()
|
| 131 |
+
torch.cuda.reset_peak_memory_stats()
|
| 132 |
sd_image = pipe(
|
| 133 |
prompt=prompt, prompt_2=prompt, prompt_3=prompt,
|
| 134 |
negative_prompt=neg_prompt_1, negative_prompt_2=neg_prompt_2, negative_prompt_3=neg_prompt_3,
|
|
|
|
| 137 |
max_sequence_length=384
|
| 138 |
).images[0]
|
| 139 |
print('-- got image --')
|
| 140 |
+
torch.cuda.empty_cache()
|
| 141 |
+
torch.cuda.reset_peak_memory_stats()
|
| 142 |
with torch.no_grad():
|
| 143 |
upscale = upscaler_2(sd_image, tiling=True, tile_width=256, tile_height=256)
|
| 144 |
upscale2 = upscaler_2(upscale, tiling=True, tile_width=256, tile_height=256)
|
|
|
|
| 151 |
seed = random.randint(0, MAX_SEED)
|
| 152 |
generator = torch.Generator(device=device).manual_seed(seed)
|
| 153 |
print('-- generating image --')
|
| 154 |
+
torch.cuda.empty_cache()
|
| 155 |
+
torch.cuda.reset_peak_memory_stats()
|
| 156 |
sd_image = pipe(
|
| 157 |
prompt=prompt, prompt_2=prompt, prompt_3=prompt,
|
| 158 |
negative_prompt=neg_prompt_1, negative_prompt_2=neg_prompt_2, negative_prompt_3=neg_prompt_3,
|
|
|
|
| 161 |
max_sequence_length=384
|
| 162 |
).images[0]
|
| 163 |
print('-- got image --')
|
| 164 |
+
torch.cuda.empty_cache()
|
| 165 |
+
torch.cuda.reset_peak_memory_stats()
|
| 166 |
with torch.no_grad():
|
| 167 |
upscale = upscaler_2(sd_image, tiling=True, tile_width=256, tile_height=256)
|
| 168 |
upscale2 = upscaler_2(upscale, tiling=True, tile_width=256, tile_height=256)
|