sagar4tech commited on
Commit
41108f2
1 Parent(s): ceee367

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ model_file_path = "/model/file/path/to/model/directory/"
11
  url = "runwayml/stable-diffusion-v1-5"
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
 
 
11
  url = "runwayml/stable-diffusion-v1-5"
12
 
13
  # Load the safetensor model
14
+ pipe = DiffusionPipeline.from_pretrained(url)
15
 
16
  #to use metal performance shader for gpu inference optimization pipe = pipe.to("mps")
17