lunarflu HF Staff commited on
Commit
54c827e
·
1 Parent(s): b2dbed9

[dfif] stage_1_results, stage_1_param_path, stage_1_result_path = await asyncio.get_running_loop().run_in_executor

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -220,8 +220,12 @@ async def deepfloydif(ctx, *, prompt: str):
220
  await ctx.message.add_reaction('❌')
221
 
222
  try:
223
- stage_1_results, stage_1_param_path, stage_1_result_path = df.predict(
224
- prompt, negative_prompt, seed, number_of_images, guidance_scale, custom_timesteps_1, number_of_inference_steps, api_name='/generate64')
 
 
 
 
225
 
226
  partialpath = stage_1_result_path[5:] #magic for later
227
  except Exception as e:
 
220
  await ctx.message.add_reaction('❌')
221
 
222
  try:
223
+ #stage_1_results, stage_1_param_path, stage_1_result_path = df.predict(
224
+ # prompt, negative_prompt, seed, number_of_images, guidance_scale, custom_timesteps_1, number_of_inference_steps, api_name='/generate64')
225
+
226
+ stage_1_results, stage_1_param_path, stage_1_result_path = await asyncio.get_running_loop().run_in_executor(
227
+ None, df.predict, prompt, negative_prompt, seed, number_of_images, guidance_scale, custom_timesteps_1, number_of_inference_steps,
228
+ api_name='/generate64')
229
 
230
  partialpath = stage_1_result_path[5:] #magic for later
231
  except Exception as e: