dongyuxuan Claude Opus 4.6 commited on
Commit
bc8fb9c
·
1 Parent(s): afc39e6

Fix torch.load compatibility and pin Python 3.10 for HF Space

Browse files

- Add weights_only=False to torch.load (required since PyTorch 2.6)
- Set python_version: "3.10" in Space config to avoid Python 3.13
compatibility issues with gradio/huggingface_hub

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

Files changed (3) hide show
  1. README.md +1 -0
  2. inference/pipeline.py +1 -1
  3. push_hf_space.sh +1 -0
README.md CHANGED
@@ -5,6 +5,7 @@ colorFrom: blue
5
  colorTo: green
6
  sdk: gradio
7
  sdk_version: "4.44.0"
 
8
  app_file: app.py
9
  pinned: false
10
  license: mit
 
5
  colorTo: green
6
  sdk: gradio
7
  sdk_version: "4.44.0"
8
+ python_version: "3.10"
9
  app_file: app.py
10
  pinned: false
11
  license: mit
inference/pipeline.py CHANGED
@@ -424,7 +424,7 @@ def _load_weights(model: torch.nn.Module,
424
  f" Error: {e}"
425
  )
426
 
427
- state_dict = torch.load(weights_path, map_location='cpu')
428
  # Filter out training-only keys (e.g. ManifoldMixup), same logic as backend.py
429
  model_dict = model.state_dict()
430
  filtered = {k: v for k, v in state_dict.items()
 
424
  f" Error: {e}"
425
  )
426
 
427
+ state_dict = torch.load(weights_path, map_location='cpu', weights_only=False)
428
  # Filter out training-only keys (e.g. ManifoldMixup), same logic as backend.py
429
  model_dict = model.state_dict()
430
  filtered = {k: v for k, v in state_dict.items()
push_hf_space.sh CHANGED
@@ -30,6 +30,7 @@ if ! git lfs version >/dev/null 2>&1; then
30
  exit 1
31
  fi
32
  git lfs install
 
33
 
34
  echo "==> Updating ignore rules..."
35
  touch .gitignore
 
30
  exit 1
31
  fi
32
  git lfs install
33
+ git config lfs.https://hf.co/spaces/yxdong0320/SELD_Streaming_RC_demo.git/info/lfs.locksverify false
34
 
35
  echo "==> Updating ignore rules..."
36
  touch .gitignore