Downloads · 30 days
0
muse-512-finetuned
amused/muse-512-finetuned
muse-512-finetuned is a machine learning model from amused. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
Access
Public
Updated Sep 19, 2023
Repo size
4.1 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin2 GB · 100%
At a glance
- Access
- Public
- Created
- Sep 19, 2023
- Updated
- Sep 19, 2023
- SHA
- 073f9bf6
Model card
From the Hugging Face model README
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]
- Created
- Sep 19, 2023
- Updated
- Sep 19, 2023