Modelpublic

linoyts/dreambooth-groot

sourceHugging Faceupdated 3y agoView on Hugging Face

1likes6downloads

Model Card

DreamBooth model for the groot concept traine on the LinoyTsaban/dreambooth-hackathon-images-groot dataset.

This is a Stable Diffusion model fine-tuned on the groot concept with DreamBooth. It can be used by modifying the instance_prompt: groot figurine

This model was created as part of the DreamBooth Hackathon 🔥. Visit the organisation page for instructions on how to take part!

params used: scheduler- DDIMScheduler resolution=512 learningrate=2e-06 maxtrainsteps= 500 trainbatchsize=1 gradientaccumulationsteps=2 maxgradnorm=1.0 gradientcheckpointing=True use8bitadam=True seed=14071995, samplebatchsize=2

Usage

python

from diffusers import StableDiffusionPipeline

pipeline = StableDiffusionPipeline.from_pretrained('LinoyTsaban/dreambooth-groot')
name_of_concept = "groot"  
type_of_thing = "figurine"  

prompt = f"a photo of groot figurine in the Louvre museum" 

# Tune the guidance to control how closely the generations follow the prompt.
# Values between 7-11 usually work best
guidance_scale = 8 # 
image = pipeline(prompt, guidance_scale=guidance_scale).images[0]
image