city535367 commited on
Commit
244122a
1 Parent(s): 8ec293e
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,7 +5,7 @@ from diffusers import StableDiffusionPipeline
5
 
6
  auth_token = "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"
7
  model_id = "CompVis/stable-diffusion-v1-4"
8
- device = "cpu"
9
 
10
  """#載入StableDiffusion model"""
11
 
@@ -130,5 +130,5 @@ with gr.Blocks() as demo:
130
 
131
 
132
  #clear_output()
133
- demo.launch()
134
 
 
5
 
6
  auth_token = "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"
7
  model_id = "CompVis/stable-diffusion-v1-4"
8
+ device = "cuda"
9
 
10
  """#載入StableDiffusion model"""
11
 
 
130
 
131
 
132
  #clear_output()
133
+ demo.launch(share=True)
134