pengdaqian commited on
Commit
8e6eef3
1 Parent(s): ca13e69
Files changed (1) hide show
  1. warm_up.py +1 -3
warm_up.py CHANGED
@@ -15,6 +15,4 @@ if __name__ == "__main__":
15
  text = pipeline.audio2txt('/home/user/app/demo.wav')
16
 
17
  text2img_pipeline = init_text2img_pipe()
18
- images = predict(text, " ".join(negative_prompt), text2img_pipeline)
19
- for idx, image in enumerate(images):
20
- image.save(f"/root/autodl-tmp/image_{idx}.png")
 
15
  text = pipeline.audio2txt('/home/user/app/demo.wav')
16
 
17
  text2img_pipeline = init_text2img_pipe()
18
+ images = predict(text, " ".join(negative_prompt), text2img_pipeline)