improvements
Browse files
app.py
CHANGED
@@ -302,7 +302,7 @@ async def deepfloydif(ctx, *, prompt: str):
|
|
302 |
number_of_inference_steps = 50
|
303 |
api_name = '/generate64'
|
304 |
|
305 |
-
await thread.send(f'
|
306 |
|
307 |
except Exception as e:
|
308 |
print(f"Error: {e}")
|
@@ -395,7 +395,8 @@ async def deepfloydif(ctx, *, prompt: str):
|
|
395 |
# Stage 2 ✅
|
396 |
async def dfif2(index: int, stage_1_result_path, thread, dfif_command_message_id): # add safetychecks
|
397 |
try:
|
398 |
-
|
|
|
399 |
|
400 |
# run blocking function in executor
|
401 |
loop = asyncio.get_running_loop()
|
|
|
302 |
number_of_inference_steps = 50
|
303 |
api_name = '/generate64'
|
304 |
|
305 |
+
await thread.send(f'{ctx.author.mention}Generating images in thread, can take ~1 minute...')
|
306 |
|
307 |
except Exception as e:
|
308 |
print(f"Error: {e}")
|
|
|
395 |
# Stage 2 ✅
|
396 |
async def dfif2(index: int, stage_1_result_path, thread, dfif_command_message_id): # add safetychecks
|
397 |
try:
|
398 |
+
number = index + 1
|
399 |
+
await thread.send(f"Upscaling number {number}...")
|
400 |
|
401 |
# run blocking function in executor
|
402 |
loop = asyncio.get_running_loop()
|