INNER CODE UNIT · Python

infer_and_save

google/break-a-scene · inference.py:53

    def infer_and_save(self, prompts):
        images = self.pipeline(prompts).images
        images[0].save(self.args.output_path)


if __name__ == "__main__":
    break_a_scene_inference = BreakASceneInference()
    break_a_scene_inference.infer_and_save(
        prompts=[break_a_scene_inference.args.prompt]
    )

View source record →