Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-21 21:58:47 +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,
)
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

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
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):

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,
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):

2
tests/adapters/test_range_adapter.py
View file

@ -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