technicalkiwi
/
latentblending
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

guidance scale fix

This commit is contained in:
lugo 2022-11-28 12:43:33 +01:00
parent bbe8269146
commit b5b703c516
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
latent_blending.py
View File

@ -69,7 +69,6 @@ class LatentBlending():
""" """
self.sdh = sdh self.sdh = sdh
self.device = self.sdh.device self.device = self.sdh.device
self.guidance_scale = guidance_scale
self.width = self.sdh.width self.width = self.sdh.width
self.height = self.sdh.height self.height = self.sdh.height
self.seed = 420 #use self.set_seed or fixed_seeds argument in run_transition self.seed = 420 #use self.set_seed or fixed_seeds argument in run_transition
@ -90,7 +89,7 @@ class LatentBlending():
self.num_inference_steps = -1 self.num_inference_steps = -1
self.list_injection_idx = None self.list_injection_idx = None
self.list_nmb_branches = None self.list_nmb_branches = None
self.set_guidance_scale(guidance_scale)
self.init_mode() self.init_mode()
@ -106,6 +105,12 @@ class LatentBlending():
else: else:
self.mode = 'standard' self.mode = 'standard'
def set_guidance_scale(self, guidance_scale):
r"""
sets the guidance scale.
"""
self.guidance_scale = guidance_scale
self.sdh.guidance_scale = guidance_scale
def set_prompt1(self, prompt: str): def set_prompt1(self, prompt: str):
r""" r"""