Hugging Face's logo Hugging Face
  • Models
  • Datasets
  • Spaces
  • Buckets new
  • Docs
  • Enterprise
  • Pricing
    • Website
      • Tasks
      • HuggingChat
      • Collections
      • Languages
      • Organizations
    • Community
      • Blog
      • Posts
      • Daily Papers
      • Learn
      • Discord
      • Forum
      • GitHub
    • Solutions
      • Team & Enterprise
      • Hugging Face PRO
      • Enterprise Support
      • Inference Providers
      • Inference Endpoints
      • Storage Buckets

  • Log In
  • Sign Up

thu-ml
/
unidiffuser-v0

Text-to-Image
Diffusers
UniDiffuserPipeline
image-to-text
image-captioning
image-variation
text-variation
multi-modality
generative model
Model card Files Files and versions
xet
Community
6

Instructions to use thu-ml/unidiffuser-v0 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

  • Libraries
  • Diffusers

    How to use thu-ml/unidiffuser-v0 with Diffusers:

    pip install -U diffusers transformers accelerate
    import torch
    from diffusers import DiffusionPipeline
    
    # switch to "mps" for apple devices
    pipe = DiffusionPipeline.from_pretrained("thu-ml/unidiffuser-v0", 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
  • Local Apps
  • Draw Things
  • DiffusionBee
unidiffuser-v0
Ctrl+K
Ctrl+K
  • 5 contributors
History: 15 commits
patrickvonplaten's picture
patrickvonplaten
dg845's picture
dg845
Upload 2 files (#5)
be59f8b over 2 years ago
  • clip_image_processor
    Upload 2 files (#5) over 2 years ago
  • clip_tokenizer
    Upload 22 files (#3) about 3 years ago
  • image_encoder
    Upload 22 files (#3) about 3 years ago
  • image_processor
    Upload 22 files (#3) about 3 years ago
  • scheduler
    Upload 22 files (#3) about 3 years ago
  • text_decoder
    Upload 22 files (#3) about 3 years ago
  • text_encoder
    Upload 22 files (#3) about 3 years ago
  • text_tokenizer
    Upload 22 files (#3) about 3 years ago
  • unet
    Upload 22 files (#3) about 3 years ago
  • vae
    Upload 22 files (#3) about 3 years ago
  • .gitattributes
    1.48 kB
    initial commit about 3 years ago
  • README.md
    7.58 kB
    Update README.md (#4) about 3 years ago
  • autoencoder_kl.pth

    Detected Pickle imports (3)

    • "collections.OrderedDict",
    • "torch._utils._rebuild_tensor_v2",
    • "torch.FloatStorage"

    What is a pickle import?

    335 MB
    xet
    autoencoder_kl.pth about 3 years ago
  • caption_decoder.pth

    Detected Pickle imports (5)

    • "collections.OrderedDict",
    • "torch.ByteStorage",
    • "torch.LongStorage",
    • "torch.FloatStorage",
    • "torch._utils._rebuild_tensor_v2"

    What is a pickle import?

    1 GB
    xet
    Upload caption_decoder.pth about 3 years ago
  • model_index.json
    699 Bytes
    Upload 2 files (#5) over 2 years ago
  • uvit_v0.pth
    3.81 GB
    xet
    uvit_v0 about 3 years ago