mirror of
https://github.com/finegrain-ai/refiners.git
synced 2024-11-09 23:12:02 +00:00
rename ResidualBlock to ConditionScaleBlock in LCM
This commit is contained in:
parent
5f21922925
commit
03b79d6d34
|
@ -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",
|
||||
)
|
||||
|
||||
|
||||
|
|
|
@ -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()
|
||||
|
|
Loading…
Reference in a new issue