Prompt-Free-Diffusion / lib /model_zoo /seecoder_decoder.py
3v324v23's picture
code pushed
515f781
raw
history blame contribute delete
402 Bytes
import numpy as np
import torch
import torch.nn as nn
import torch.nn.functional as F
from torch.nn.init import xavier_uniform_, constant_, uniform_, normal_
from lib.model_zoo.common.get_model import get_model, register
from .seecoder_utils import PositionEmbeddingSine, _get_clones, \
_get_activation_fn, _is_power_of_2, c2_xavier_fill, Conv2d_Convenience
###########
# modules #
###########