Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-24 15:18:46 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

add pixel unshuffle to fluxion's layers

Browse source
This commit is contained in:
Cédric Deltheil 2023-09-24 14:43:38 +02:00 committed by Cédric Deltheil
parent 251277a0a8
commit 4352e78483
2 changed files with 9 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/refiners/fluxion/layers/__init__.py
View file

@ -40,6 +40,7 @@ from refiners.fluxion.layers.conv import Conv2d, ConvTranspose2d
from refiners.fluxion.layers.linear import Linear, MultiLinear from refiners.fluxion.layers.linear import Linear, MultiLinear
from refiners.fluxion.layers.module import Module, WeightedModule, ContextModule from refiners.fluxion.layers.module import Module, WeightedModule, ContextModule
from refiners.fluxion.layers.padding import ReflectionPad2d from refiners.fluxion.layers.padding import ReflectionPad2d
from refiners.fluxion.layers.pixelshuffle import PixelUnshuffle
from refiners.fluxion.layers.sampling import Downsample, Upsample, Interpolate from refiners.fluxion.layers.sampling import Downsample, Upsample, Interpolate
from refiners.fluxion.layers.embedding import Embedding from refiners.fluxion.layers.embedding import Embedding
from refiners.fluxion.layers.converter import Converter from refiners.fluxion.layers.converter import Converter
@ -101,5 +102,6 @@ __all__ = [
"ContextModule", "ContextModule",
"Interpolate", "Interpolate",
"ReflectionPad2d", "ReflectionPad2d",
"PixelUnshuffle",
"Converter", "Converter",
] ]

7
src/refiners/fluxion/layers/pixelshuffle.py Normal file
View file

@ -0,0 +1,7 @@
from refiners.fluxion.layers.module import Module
from torch.nn import PixelUnshuffle as _PixelUnshuffle
class PixelUnshuffle(_PixelUnshuffle, Module):
def __init__(self, downscale_factor: int):
_PixelUnshuffle.__init__(self, downscale_factor=downscale_factor)