Untitled

Anonymous
python
05/09/2023 9:43 PM
368 B
17
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)
Editor is loading...