Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-25 07:38:45 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

scope range adapter in latent diffusion

Browse source
This commit is contained in:
Pierre Chapuis 2023-09-01 16:45:28 +02:00
parent 73813310d0
commit 31785f2059
5 changed files with 8 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

0
src/refiners/adapters/range_adapter.py → src/refiners/foundationals/latent_diffusion/range_adapter.py
View file

2
src/refiners/foundationals/latent_diffusion/stable_diffusion_1/controlnet.py
View file

@ -8,7 +8,7 @@ from refiners.foundationals.latent_diffusion.stable_diffusion_1.unet import (
TimestepEncoder, TimestepEncoder,
) )
from refiners.adapters.adapter import Adapter 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 typing import cast, Iterable
from torch import Tensor, device as Device, dtype as DType from torch import Tensor, device as Device, dtype as DType

2
src/refiners/foundationals/latent_diffusion/stable_diffusion_1/unet.py
View file

@ -6,7 +6,7 @@ from refiners.fluxion.context import Contexts
import refiners.fluxion.layers as fl import refiners.fluxion.layers as fl
from refiners.foundationals.latent_diffusion.cross_attention import CrossAttentionBlock2d 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): class TimestepEncoder(fl.Passthrough):

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

@ -8,7 +8,11 @@ from refiners.foundationals.latent_diffusion.stable_diffusion_1.unet import (
ResidualBlock, ResidualBlock,
ResidualConcatenator, 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): class TextTimeEmbedding(fl.Chain):

2
tests/adapters/test_range_adapter.py
View file

@ -1,6 +1,6 @@
import torch import torch
from refiners.adapters.adapter import Adapter 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 from refiners.fluxion.layers import Chain, Linear