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