[deepfloyd.py] testing dfif2 job.submit
Browse files- 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)"""
|