Views
No views yet
1pip install transformers accelerate ftfy
2pip install -U git+https://github.com/huggingface/open-muse.git1import torch
2from muse import PipelineMuse
3
4pipe = PipelineMuse.from_pretrained("openMUSE/muse-512-finetuned").to("cuda", dtype=torch.float16)
5pipe.transformer.enable_xformers_memory_efficient_attention()
6
7prompt = "A monkey doing yoga on the beach"
8images = pipe(
9 prompt, timesteps=16, guidance_scale=10, transformer_seq_len=1024, use_fp16=True
10)
11images[0]