Skip to content

Instantly share code, notes, and snippets.

@isozu

isozu/sd_initial_setup.py Secret

Created August 30, 2022 05:26
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save isozu/05e20428bab160ddb76bed159cb4fa0a to your computer and use it in GitHub Desktop.
Save isozu/05e20428bab160ddb76bed159cb4fa0a to your computer and use it in GitHub Desktop.
Download ZIP
Raw
sd_initial_setup.py
#@title Initial Setup
!pip install diffusers==0.2.4 transformers scipy ftfy
ACCESS_TOKEN="{{YOUR_TOKEN}}"#@param {type:"string"}
from diffusers import StableDiffusionPipeline
pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=ACCESS_TOKEN)
pipe.to("cuda")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment