fire23's starred repositories
pytorch-OpCounter
Count the MACs / FLOPs of your PyTorch model.
latent-consistency-model
Latent Consistency Models: Synthesizing High-Resolution Images with Few-Step Inference
celeba-hq-modified
Modified h5tool.py make user getting celeba-HQ easier
make-CelebA-HQ
Supposed you've downloaded CelebA & CelebA-HQ dataset, and want to get HQ images from them.
progressive_growing_of_gans
Progressive Growing of GANs for Improved Quality, Stability, and Variation
torch-fidelity
High-fidelity performance metrics for generative models in PyTorch
PixelCNN-Pytorch
A naive implementation of PixelCNN in Pytorch as described in A Oord et. al.
pytorchviz
A small package to create visualizations of PyTorch execution graphs
SwissRoll-VAE-pytorch
VAE model for approximating the distribution of the Swiss roll dataset.
PyTorch-VAE
A Collection of Variational Autoencoders (VAE) in PyTorch.
enhancing-transformers
An unofficial implementation of both ViT-VQGAN and RQ-VAE in Pytorch
Ask-Anything
[CVPR2024 Highlight][VideoChatGPT] ChatGPT with video understanding! And many more supported LMs such as miniGPT4, StableLM, and MOSS.
imagen-pytorch
Implementation of Imagen, Google's Text-to-Image Neural Network, in Pytorch
generators-with-stylegan2
Here is a series of face generators based on StyleGAN2
tryondiffusion
PyTorch implementation of "TryOnDiffusion: A Tale of Two UNets", a virtual try-on diffusion-based network by Google
InternVideo
Video Foundation Models & Data for Multimodal Understanding