1
0
Fork 0

fixes for optimum

This commit is contained in:
Sean Sube 2023-08-20 22:29:07 -05:00
parent 404a314050
commit ee430a962b
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 2 additions and 2 deletions

View File

@ -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,

View File

@ -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: