lunarflu HF Staff commited on
Commit
64a29e0
·
1 Parent(s): 524f380

[deepfloydif.py] df -> deepfloyd_client

Browse files
Files changed (1) hide show
  1. 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 = df.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]
 
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]