jasperai
/

LBM_depth

Model card Files Files and versions

clementchadebec commited on Jul 10

Commit

764ca30

·

verified ·

1 Parent(s): d589e2b

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -26,7 +26,7 @@ from lbm.inference import evaluate, get_model
 # Load model
 model = get_model(
-  "jasperai/LBM_relighting",
   torch_dtype=torch.bfloat16,
   device="cuda",
 )

 # Load model
 model = get_model(
+  "jasperai/LBM_depth",
   torch_dtype=torch.bfloat16,
   device="cuda",
 )