SceneCraft: Layout-Guided 3D Scene Generation
Paper • 2410.09049 • Published • 3
How to use gzzyyxy/layout_diffusion_hypersim_prompt_one_hot_multi_control_bs32_epoch24 with Diffusers:
pip install -U diffusers transformers accelerate
import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("gzzyyxy/layout_diffusion_hypersim_prompt_one_hot_multi_control_bs32_epoch24", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]This repository contains the model described in the paper SceneCraft: Layout-Guided 3D Scene Generation.
Project page: https://orangesodahub.github.io/SceneCraft