Untitled

 avatar
unknown
python
2 years ago
275 B
4
Indexable
# put it into folder: pretrained 

import torch
from diffusers import StableDiffusionPipeline

model_id = "runwayml/stable-diffusion-v1-5"
path = "."
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
pipe.save_pretrained(save_directory=path)