imaginAIry/imaginairy/vendored/refiners/fluxion/layers/padding.py
Bryce 55e27160f5 build: vendorize refiners
so we can still work in conda envs
2024-01-02 22:02:31 -08:00

9 lines
233 B
Python

from torch import nn
from imaginairy.vendored.refiners.fluxion.layers.module import Module
class ReflectionPad2d(nn.ReflectionPad2d, Module):
def __init__(self, padding: int) -> None:
super().__init__(padding=padding)