StableDesign

Running on Zero

nick_93 commited on Apr 7

Commit

f732e49

•

1 Parent(s): ab3659a

init

Files changed (1) hide show

app.py CHANGED Viewed

@@ -112,7 +112,7 @@ def segment_image(
 def get_depth_pipeline():
     feature_extractor = AutoImageProcessor.from_pretrained("LiheYoung/depth-anything-large-hf",
-                                                           torch_dtype=dtype).to(device)
     depth_estimator = AutoModelForDepthEstimation.from_pretrained("LiheYoung/depth-anything-large-hf",
                                                                   torch_dtype=dtype).to(device)
     return feature_extractor, depth_estimator

 def get_depth_pipeline():
     feature_extractor = AutoImageProcessor.from_pretrained("LiheYoung/depth-anything-large-hf",
+                                                           torch_dtype=dtype)
     depth_estimator = AutoModelForDepthEstimation.from_pretrained("LiheYoung/depth-anything-large-hf",
                                                                   torch_dtype=dtype).to(device)
     return feature_extractor, depth_estimator