How to use keras-dreambooth/dreambooth-pug-ace 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-pug-ace")
How to use keras-dreambooth/dreambooth-pug-ace 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-pug-ace", dtype=torch.bfloat16, device_map="cuda") prompt = "a photo of puggieace dog on the beach, sunset in background" image = pipe(prompt).images[0]
