Merge branch 'main' of https://huggingface.co/spaces/Stable-X/normal-estimation-comparison

app.py

@@ -46,7 +46,7 @@ class Geowizard(object):
     '''
 
     def __init__(self):
-        self.model = DepthNormalEstimationPipeline.from_pretrained("weights/Geowizard/", torch_dtype=torch.float16)
+        self.model = DepthNormalEstimationPipeline.from_pretrained("lemonaddie/Geowizard", torch_dtype=torch.float16)
 
     def cuda(self):
        self.model.cuda()
@@ -103,7 +103,7 @@ class Marigold(Geowizard):
     '''
 
     def __init__(self):
-        self.model= diffusers.MarigoldNormalsPipeline.from_pretrained("weights/marigold-normals-v0-1", torch_dtype=torch.float16)
+        self.model= diffusers.MarigoldNormalsPipeline.from_pretrained("prs-eth/marigold-normals-v0-1", torch_dtype=torch.float16)
 
 
     @torch.no_grad()
@@ -131,9 +131,9 @@ class StableNormal(Geowizard):
     '''
 
     def __init__(self):
-        x_start_pipeline = YOSONormalsPipeline.from_pretrained('/workspace/code/InverseRendering/StableNormal/weights/yoso-normal-v0-2', 
+        x_start_pipeline = YOSONormalsPipeline.from_pretrained('Stable-X/yoso-normal-v0-2',  trust_remote_code=True,
                                                                 variant="fp16", torch_dtype=torch.float16)
-        self.model = StableNormalPipeline.from_pretrained('/workspace/code/InverseRendering/StableNormal/weights/stable-normal-v0-1',
+        self.model = StableNormalPipeline.from_pretrained('Stable-X/stable-normal-v0-1', trust_remote_code=True,
                                                            variant="fp16",  torch_dtype=torch.float16,
                                                           scheduler=HEURI_DDIMScheduler(prediction_type='sample', 
                                                                               beta_start=0.00085, beta_end=0.0120,