File size: 357 Bytes
188fd89 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
#!/usr/bin/env python
import os
import sys
import torch
from diffusers import StableDiffusionPipeline
os.makedirs("diffusers-cache", exist_ok=True)
pipe = StableDiffusionPipeline.from_pretrained(
"CompVis/stable-diffusion-v1-4",
cache_dir="diffusers-cache",
revision="fp16",
torch_dtype=torch.float16,
use_auth_token=sys.argv[1],
)
|