1from PIL import Image
2import torch
3from muse import PipelineMuse, MaskGiTUViT
4from datasets import Dataset, Features
5from datasets import Image as ImageFeature
6from datasets import Value, load_dataset
7
8device = "cuda" if torch.cuda.is_available() else "cpu"
9
10pipe = PipelineMuse.from_pretrained(
11 transformer_path="valhalla/research-run",
12 text_encoder_path="openMUSE/clip-vit-large-patch14-text-enc"… See the full description on the dataset page: https://huggingface.co/datasets/diffusers-parti-prompts/muse512.