[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)"""
|