mirror of
https://github.com/brycedrennan/imaginAIry
synced 2024-10-31 03:20:40 +00:00
55e27160f5
so we can still work in conda envs
9 lines
233 B
Python
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)
|