lunarflu HF Staff commited on
Commit
20c6e2e
·
1 Parent(s): 76e8bd7

[deepfloyd.py] testing dfif2 job.submit

Browse files
Files changed (1) hide show
  1. deepfloydif.py +10 -1
deepfloydif.py CHANGED
@@ -39,9 +39,18 @@ def deepfloydif_stage_2_inference(index, stage_1_result_path):
39
  guidance_scale_2 = 4
40
  custom_timesteps_2 = 'smart50'
41
  number_of_inference_steps_2 = 50
 
 
 
 
 
 
 
42
  result_path = deepfloydif_client.predict(stage_1_result_path, selected_index_for_stage_2, seed_2, guidance_scale_2, custom_timesteps_2, number_of_inference_steps_2, api_name='/upscale256')
43
  print(result_path)
44
- return result_path
 
 
45
 
46
  async def deepfloydif_stage_1(interaction, prompt, client):
47
  """DeepfloydIF command (generate images with realistic text using slash commands)"""
 
39
  guidance_scale_2 = 4
40
  custom_timesteps_2 = 'smart50'
41
  number_of_inference_steps_2 = 50
42
+
43
+ job = deepfloydif_client.submit(stage_1_result_path, selected_index_for_stage_2, seed_2, guidance_scale_2, custom_timesteps_2, number_of_inference_steps_2, api_name='/upscale256')
44
+ result_path = job.outputs()
45
+ return result_path
46
+
47
+
48
+ '''
49
  result_path = deepfloydif_client.predict(stage_1_result_path, selected_index_for_stage_2, seed_2, guidance_scale_2, custom_timesteps_2, number_of_inference_steps_2, api_name='/upscale256')
50
  print(result_path)
51
+ return result_path
52
+ '''
53
+
54
 
55
  async def deepfloydif_stage_1(interaction, prompt, client):
56
  """DeepfloydIF command (generate images with realistic text using slash commands)"""