Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-09-19 19:05:28 +00:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

rename ResidualBlock to ConditionScaleBlock in LCM

Browse source
This commit is contained in:
Pierre Chapuis 2024-02-21 15:53:58 +01:00
parent 5f21922925
commit 03b79d6d34
2 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
scripts/prepare_test_weights.py
View file

@ -678,7 +678,7 @@ def convert_lcm_base():
"tests/weights/latent-consistency/lcm-sdxl",
"tests/weights/sdxl-lcm-unet.safetensors",
half=True,
expected_hash="242cf440",
expected_hash="e161b20c",
)

6
src/refiners/foundationals/latent_diffusion/stable_diffusion_xl/lcm.py
View file

@ -29,7 +29,7 @@ def compute_sinusoidal_embedding(
return embedding
class ResidualBlock(fl.Residual):
class ConditionScaleBlock(fl.Residual):
def __init__(
self,
in_channels: int,
@ -85,7 +85,7 @@ class SDXLLcmAdapter(fl.Chain, Adapter[SDXLUNet]):
def inject(self: "SDXLLcmAdapter", parent: fl.Chain | None = None) -> "SDXLLcmAdapter":
ra = self.target.ensure_find(RangeEncoder)
block = ResidualBlock(
block = ConditionScaleBlock(
in_channels=self.condition_scale_embedding_dim,
out_channels=ra.sinusoidal_embedding_dim,
device=self.target.device,
@ -96,5 +96,5 @@ class SDXLLcmAdapter(fl.Chain, Adapter[SDXLUNet]):
def eject(self) -> None:
ra = self.target.ensure_find(RangeEncoder)
ra.remove(ra.ensure_find(ResidualBlock))
ra.remove(ra.ensure_find(ConditionScaleBlock))
super().eject()