Instructions to use herve76/bb with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use herve76/bb with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", dtype=torch.bfloat16, device_map="cuda") pipe.load_lora_weights("herve76/bb") prompt = "bbhf" image = pipe(prompt).images[0] - Inference
- Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- Draw Things
- DiffusionBee
| base_model: stabilityai/stable-diffusion-xl-base-1.0 | |
| instance_prompt: bbhf | |
| tags: | |
| - stable-diffusion-xl | |
| - stable-diffusion-xl-diffusers | |
| - text-to-image | |
| - diffusers | |
| - lora | |
| inference: false | |
| datasets: | |
| - herve76/bb | |
| # LoRA DreamBooth - herve76/bb | |
| These are LoRA adaption weights for stabilityai/stable-diffusion-xl-base-1.0. | |
| The weights were trained on the concept prompt: | |
| ``` | |
| bbhf | |
| ``` | |
| Use this keyword to trigger your custom model in your prompts. | |
| LoRA for the text encoder was enabled: False. | |
| Special VAE used for training: madebyollin/sdxl-vae-fp16-fix. | |
| ## Usage | |
| Make sure to upgrade diffusers to >= 0.19.0: | |
| ``` | |
| pip install diffusers --upgrade | |
| ``` | |
| In addition make sure to install transformers, safetensors, accelerate as well as the invisible watermark: | |
| ``` | |
| pip install invisible_watermark transformers accelerate safetensors | |
| ``` | |
| To just use the base model, you can run: | |
| ```python | |
| import torch | |
| from diffusers import DiffusionPipeline, AutoencoderKL | |
| vae = AutoencoderKL.from_pretrained('madebyollin/sdxl-vae-fp16-fix', torch_dtype=torch.float16) | |
| pipe = DiffusionPipeline.from_pretrained( | |
| "stabilityai/stable-diffusion-xl-base-1.0", | |
| vae=vae, torch_dtype=torch.float16, variant="fp16", | |
| use_safetensors=True | |
| ) | |
| pipe.to("cuda") | |
| # This is where you load your trained weights | |
| pipe.load_lora_weights('herve76/bb') | |
| prompt = "A majestic bbhf jumping from a big stone at night" | |
| image = pipe(prompt=prompt, num_inference_steps=50).images[0] | |
| ``` | |