import torch
from diffusers import StableDiffusionPipeline
model_id = "SG161222/Realistic_Vision_V2.0"

pipeline = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
pipeline = pipeline.to("cuda")

prompt = "floor"
negative_prompt = "person, sofa, chair"

image = pipeline(prompt, negative_prompt=negative_prompt, guidance_scale=7.5, \
			num_inference_steps=50, height=384, width=512).images[0]
image.save("./genFloor/0.jpg")
image