Spaces:
Sleeping
Sleeping
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>
- README.md +1 -0
- inference/pipeline.py +1 -1
- 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
|