- self.generator.manual_seed(self.gen_params.seed)
- image = self.pipe(generator=self.generator,
- prompt=self.gen_params.prompt,
- guidance_scale=self.gen_params.guidance,
- height=self.gen_params.height,
- width=self.gen_params.width,
- num_inference_steps=self.gen_params.n_steps,
- ).images[0]
+ self._generator.manual_seed(self._gen_params.seed)
+ image = self._pipe(generator=self._generator,
+ prompt=self._gen_params.prompt,
+ guidance_scale=self._gen_params.guidance,
+ height=self._gen_params.height,
+ width=self._gen_params.width,
+ num_inference_steps=self._gen_params.n_steps,
+ ).images[0]