[deepfloydif.py] df -> deepfloyd_client
Browse files- deepfloydif.py +1 -1
deepfloydif.py
CHANGED
|
@@ -30,7 +30,7 @@ def inference(prompt):
|
|
| 30 |
custom_timesteps_1 = 'smart50'
|
| 31 |
number_of_inference_steps = 50
|
| 32 |
|
| 33 |
-
stage_1_results, stage_1_param_path, stage_1_result_path =
|
| 34 |
prompt, negative_prompt, seed, number_of_images, guidance_scale, custom_timesteps_1, number_of_inference_steps, api_name='/generate64')
|
| 35 |
|
| 36 |
return [stage_1_results, stage_1_param_path, stage_1_result_path]
|
|
|
|
| 30 |
custom_timesteps_1 = 'smart50'
|
| 31 |
number_of_inference_steps = 50
|
| 32 |
|
| 33 |
+
stage_1_results, stage_1_param_path, stage_1_result_path = deepfloyd_client.predict(
|
| 34 |
prompt, negative_prompt, seed, number_of_images, guidance_scale, custom_timesteps_1, number_of_inference_steps, api_name='/generate64')
|
| 35 |
|
| 36 |
return [stage_1_results, stage_1_param_path, stage_1_result_path]
|