sagar4tech commited on
Commit
ceee367
1 Parent(s): 31d5745

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -12,7 +12,8 @@ url = "runwayml/stable-diffusion-v1-5"
12
 
13
  # Load the safetensor model
14
  pipe = DiffusionPipeline.from_pretrained(url, torch_dtype=torch.float16)
15
- pipe = pipe.to("mps")
 
16
 
17
  def predict(text):
18
  # Ensure pipe(text) returns the correct output format
 
12
 
13
  # Load the safetensor model
14
  pipe = DiffusionPipeline.from_pretrained(url, torch_dtype=torch.float16)
15
+
16
+ #to use metal performance shader for gpu inference optimization pipe = pipe.to("mps")
17
 
18
  def predict(text):
19
  # Ensure pipe(text) returns the correct output format