How to use keras-dreambooth/dreambooth_diffusion_clay_cups with Keras:
# Available backend options are: "jax", "torch", "tensorflow". import os os.environ["KERAS_BACKEND"] = "jax" import keras model = keras.saving.load_model("hf://keras-dreambooth/dreambooth_diffusion_clay_cups")
How to use keras-dreambooth/dreambooth_diffusion_clay_cups with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("keras-dreambooth/dreambooth_diffusion_clay_cups", dtype=torch.bfloat16, device_map="cuda") prompt = "a photo of traditional cups image in bng clay style" image = pipe(prompt).images[0]
