Load stable diffusion pipeline with model cache in google drive. Use in Colab

load_stable_diffusion_pipeline.py

cache_dir = "/root/.cache/huggingface/hub"
!ln -fs /content/drive/MyDrive/huggingface_hub_cache $cache_dir

from diffusers import StableDiffusionPipeline
import torch

model_id = "runwayml/stable-diffusion-v1-5"
pipe = StableDiffusionPipeline.from_pretrained(model_id, cache_dir=cache_dir, torch_dtype=torch.float16)
pipe = pipe.to("cuda")

prompt = "a photo of an astronaut riding a horse on mars"
image = pipe(prompt).images[0]  
    
image.save("astronaut_rides_horse.png")
image