HBDing commited on
Commit
793189d
·
1 Parent(s): 23b2908
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from diffusers import StableDiffusionPipeline
4
  import os
5
  import torch
6
 
7
-
8
  # Load model
9
  pipe = StableDiffusionPipeline.from_pretrained(
10
  "runwayml/stable-diffusion-v1-5",
 
4
  import os
5
  import torch
6
 
7
+ import spaces #[uncomment to use ZeroGPU]
8
  # Load model
9
  pipe = StableDiffusionPipeline.from_pretrained(
10
  "runwayml/stable-diffusion-v1-5",