mirror of
https://github.com/finegrain-ai/refiners.git
synced 2024-11-21 21:58:47 +00:00
parent
0024191c58
commit
ac631bfb2e
|
@ -60,8 +60,7 @@ class StableDiffusion_1(LatentDiffusionModel):
|
|||
if sag := self._find_sag_adapter():
|
||||
sag.scale = scale
|
||||
else:
|
||||
sag = SD1SAGAdapter(target=self.unet, scale=scale)
|
||||
sag.inject()
|
||||
SD1SAGAdapter(target=self.unet, scale=scale).inject()
|
||||
else:
|
||||
if sag := self._find_sag_adapter():
|
||||
sag.eject()
|
||||
|
|
|
@ -101,8 +101,7 @@ class StableDiffusion_XL(LatentDiffusionModel):
|
|||
if sag := self._find_sag_adapter():
|
||||
sag.scale = scale
|
||||
else:
|
||||
sag = SDXLSAGAdapter(target=self.unet, scale=scale)
|
||||
sag.inject()
|
||||
SDXLSAGAdapter(target=self.unet, scale=scale).inject()
|
||||
else:
|
||||
if sag := self._find_sag_adapter():
|
||||
sag.eject()
|
||||
|
|
Loading…
Reference in a new issue