amused
/

muse-512-finetuned

Model card Files Files and versions Community

valhalla commited on Sep 19, 2023

Commit

073f9bf

•

1 Parent(s): d3fae95

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -13,7 +13,7 @@ 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"

 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"