Merge branch 'main' of https://huggingface.co/spaces/cherubicxn/ScaleLSD

app.py

@@ -32,7 +32,8 @@ def stop_run():
         gr.update(visible=False),
     )
 
-@spaces.GPU()
+# @spaces.GPU()
+@spaces.GPU
 def process_image(
     input_image,
     model_name='scalelsd-vitbase-v2-train-sa1b.pt',

scalelsd/ssl/misc/train_utils.py

@@ -46,7 +46,7 @@ def load_scalelsd_model(ckpt_path, device='cuda'):
 
     model = ScaleLSD(gray_scale=True, use_layer_scale=use_layer_scale)
     model = model.eval().to(device)
-    state_dict = torch.load(ckpt_path, map_location='cpu',weights_only=True)
+    state_dict = torch.load(ckpt_path, map_location='cpu', weights_only=True)
     try:
         model.load_state_dict(state_dict['model_state'])
     except: