Hugging Face's logo Hugging Face
  • Models
  • Datasets
  • Spaces
  • Buckets new
  • Docs
  • Enterprise
  • Pricing

  • Log In
  • Sign Up

SMILE-project
/
OpenVAE

Diffusers
Safetensors
medical
computed-tomography
vae
diffusion
Model card Files Files and versions
xet
Community

Instructions to use SMILE-project/OpenVAE with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

  • Libraries
  • Diffusers

    How to use SMILE-project/OpenVAE with Diffusers:

    pip install -U diffusers transformers accelerate
    import torch
    from diffusers import DiffusionPipeline
    
    # switch to "mps" for apple devices
    pipe = DiffusionPipeline.from_pretrained("SMILE-project/OpenVAE", dtype=torch.bfloat16, device_map="cuda")
    
    prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
    image = pipe(prompt).images[0]
  • Notebooks
  • Google Colab
  • Kaggle

You need to agree to share your contact information to access this model

This repository is publicly accessible, but you have to accept the conditions to access its files and content.

Log in or Sign Up to review the conditions and access this model content.

Gated model
You can list files but not access them

Preview of files found in this repository
  • MAISI
    Add MAISI pretrained autoencoder (MONAI MAISI) about 1 month ago
  • OpenVAE-2D-4x-100K
    2D KL VAE, trained on 100K CT data about 2 months ago
  • OpenVAE-2D-4x-20K
    2D KL VAE, trained on 20K CT data about 2 months ago
  • OpenVAE-2D-4x-300K
    2D KL VAE, trained on 300K CT data about 2 months ago
  • OpenVAE-2D-4x-PCCT_enhanced
    2D KL VAE + Discriminator, small KL penalty, trained on purely Photon Counting CT data, kuma trained, aims to preseve more details about 2 months ago
  • OpenVAE-3D-4x-PCCT_enhanced
    Upload OpenVAE-3D-4x-PCCT_enhanced/autoencoder_best.pt about 1 month ago
  • OpenVAE-3D-4x-patch64-10K
    Add OpenVAE-3D-4x-patch64-10K autoencoder_latest about 1 month ago
  • docs
    Upload 3 files 14 days ago
  • .gitattributes
    1.64 kB
    Add Anthropic-style benchmark figure (SSIM/PSNR/LPIPS bars) 20 days ago
  • README.md
    9.76 kB
    Update README.md 7 days ago