dgoot commited on
Commit
1b52f82
·
verified ·
1 Parent(s): f9d0c71

Update app.py

Browse files

Apply optimizations

Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -20,7 +20,12 @@ def load_pipeline(model):
20
  else AutoPipelineForImage2Image
21
  )
22
 
23
- return pipeline_type.from_pretrained(model)
 
 
 
 
 
24
 
25
 
26
  load_pipeline(DEFAULT_MODEL).to("cuda")
 
20
  else AutoPipelineForImage2Image
21
  )
22
 
23
+ return pipeline_type.from_pretrained(
24
+ model,
25
+ torch_dtype=torch.float16,
26
+ use_safetensors=True,
27
+ variant="fp16"
28
+ )
29
 
30
 
31
  load_pipeline(DEFAULT_MODEL).to("cuda")