Edit model card

Texture Diffusion

This DreamBooth model is fine-tuned for diffuse textures. It produces flat textures with very little visible lighting/shadows.


Here are a few example images (generated with 50 steps).

pbr uneven stone wall pbr dirt with weeds pbr bright white marble


Use the token pbr in your prompts to invoke the style.

This model was made for use in Dream Textures, a Stable Diffusion add-on for Blender.

You can also use it with 🧨 diffusers:

from diffusers import StableDiffusionPipeline
import torch

model_id = "dream-textures/texture-diffusion"

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

prompt = "pbr brick wall"
image = pipe(prompt).images[0]  

Training Details

  • Base Model: stabilityai/stable-diffusion-2-base
  • Resolution: 512
  • Prior Loss Weight: 1.0
  • Class Prompt: texture
  • Batch Size: 1
  • Learning Rate: 1e-6
  • Precision: fp16
  • Steps: 4000
  • GPU: Tesla T4


This model was trained on 278 CC0 textures from PolyHaven.

Downloads last month

Spaces using dream-textures/texture-diffusion 2