lemonaddie commited on
Commit
49fe562
1 Parent(s): e88bc8b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -383,7 +383,7 @@ def main():
383
  from pipeline.depth_normal_pipeline_clip_cfg import DepthNormalEstimationPipeline
384
 
385
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
386
- pipe = DepthNormalEstimationPipeline.from_pretrained(CHECKPOINT)
387
 
388
  try:
389
  import xformers
 
383
  from pipeline.depth_normal_pipeline_clip_cfg import DepthNormalEstimationPipeline
384
 
385
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
386
+ pipe = DepthNormalEstimationPipeline.from_pretrained(CHECKPOINT, use_safetensors=False)
387
 
388
  try:
389
  import xformers