mirror of
https://github.com/finegrain-ai/refiners.git
synced 2024-11-09 23:12:02 +00:00
scope range adapter in latent diffusion
This commit is contained in:
parent
73813310d0
commit
31785f2059
|
@ -8,7 +8,7 @@ from refiners.foundationals.latent_diffusion.stable_diffusion_1.unet import (
|
|||
TimestepEncoder,
|
||||
)
|
||||
from refiners.adapters.adapter import Adapter
|
||||
from refiners.adapters.range_adapter import RangeAdapter2d
|
||||
from refiners.foundationals.latent_diffusion.range_adapter import RangeAdapter2d
|
||||
from typing import cast, Iterable
|
||||
from torch import Tensor, device as Device, dtype as DType
|
||||
|
||||
|
|
|
@ -6,7 +6,7 @@ from refiners.fluxion.context import Contexts
|
|||
import refiners.fluxion.layers as fl
|
||||
|
||||
from refiners.foundationals.latent_diffusion.cross_attention import CrossAttentionBlock2d
|
||||
from refiners.adapters.range_adapter import RangeEncoder, RangeAdapter2d
|
||||
from refiners.foundationals.latent_diffusion.range_adapter import RangeEncoder, RangeAdapter2d
|
||||
|
||||
|
||||
class TimestepEncoder(fl.Passthrough):
|
||||
|
|
|
@ -8,7 +8,11 @@ from refiners.foundationals.latent_diffusion.stable_diffusion_1.unet import (
|
|||
ResidualBlock,
|
||||
ResidualConcatenator,
|
||||
)
|
||||
from refiners.adapters.range_adapter import RangeAdapter2d, RangeEncoder, compute_sinusoidal_embedding
|
||||
from refiners.foundationals.latent_diffusion.range_adapter import (
|
||||
RangeAdapter2d,
|
||||
RangeEncoder,
|
||||
compute_sinusoidal_embedding,
|
||||
)
|
||||
|
||||
|
||||
class TextTimeEmbedding(fl.Chain):
|
||||
|
|
|
@ -1,6 +1,6 @@
|
|||
import torch
|
||||
from refiners.adapters.adapter import Adapter
|
||||
from refiners.adapters.range_adapter import RangeEncoder
|
||||
from refiners.foundationals.latent_diffusion.range_adapter import RangeEncoder
|
||||
from refiners.fluxion.layers import Chain, Linear
|
||||
|
||||
|
||||
|
|
Loading…
Reference in a new issue