|
# MUSE-512-finetuned |
|
|
|
## Example |
|
|
|
First install the deps using |
|
|
|
```bash |
|
pip install transformers accelerate ftfy |
|
pip install -U git+https://github.com/huggingface/open-muse.git |
|
``` |
|
|
|
```python |
|
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] |
|
``` |