fixes for optimum
This commit is contained in:
parent
404a314050
commit
ee430a962b
|
@ -106,7 +106,7 @@ class SourceTxt2ImgStage(BaseStage):
|
|||
width=latent_size.width,
|
||||
generator=rng,
|
||||
guidance_scale=params.cfg,
|
||||
latents=latents,
|
||||
# latents=latents,
|
||||
negative_prompt=negative_prompt,
|
||||
num_images_per_prompt=params.batch,
|
||||
num_inference_steps=params.steps,
|
||||
|
|
|
@ -140,7 +140,7 @@ class DeviceParams:
|
|||
if self.options is None:
|
||||
return self.provider
|
||||
else:
|
||||
return (self.provider, self.options)
|
||||
return self.provider # (self.provider, self.options)
|
||||
|
||||
def sess_options(self, cache=True) -> SessionOptions:
|
||||
if cache and self.sess_options_cache is not None:
|
||||
|
|
Loading…
Reference in New Issue