Edit model card
YAML Metadata Warning: empty or missing yaml metadata in repo card (https://huggingface.co/docs/hub/model-cards#model-card-metadata)

MUSE-512-finetuned

Example

First install the deps using

pip install transformers accelerate ftfy
pip install -U git+https://github.com/huggingface/open-muse.git
import torch
from muse import PipelineMuse

pipe = PipelineMuse.from_pretrained("openMUSE/muse-512-finetuned").to("cuda", dtype=torch.float16)
pipe.transformer.enable_xformers_memory_efficient_attention()

prompt = "A monkey doing yoga on the beach"
images = pipe(
  prompt, timesteps=16, guidance_scale=10, transformer_seq_len=1024, use_fp16=True
)
images[0]
Downloads last month

-

Downloads are not tracked for this model. How to track
Inference API
Unable to determine this model's library. Check the docs .