Merge branch 'main' of https://github.com/compphoto/IntrinsicCompositing

into main
compphoto · Jan 3, 2024 · e1021ef · e1021ef
2 parents fbd119e + cdd5119
commit e1021ef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/intrinsic_compositing/shading/pipeline.py b/intrinsic_compositing/shading/pipeline.py
@@ -13,7 +13,7 @@ def load_reshading_model(path, device='cuda'):
 
     if path == 'paper_weights':
         state_dict = torch.hub.load_state_dict_from_url('https://github.com/compphoto/IntrinsicCompositing/releases/download/1.0.0/shading_paper_weights.pt', map_location=device, progress=True)
-    if path == 'further_trained'
+    if path == 'further_trained':
         state_dict = torch.hub.load_state_dict_from_url('https://github.com/compphoto/IntrinsicCompositing/releases/download/1.0.0/further_trained.pt', map_location=device, progress=True)
     else:
         state_dict = torch.load(path)