Bug on downloading model

[cindity]

Hi,
Thank you for your awesome work.

I use this code, but it shows me this error. Do you have it on your side?

import torch

predictor = torch.hub.load("unilight/sheet:v0.2.5", "sheet_ssqa",
                           trust_repo=True, force_reload=False, _id="bvcc/sslmos-wavlm_large/1337")

Downloading: "https://github.com/unilight/sheet/zipball/v0.2.5" to /root/.cache/torch/hub/v0.2.5.zip
Traceback (most recent call last):
  File "/code/fwd-local-scripts/sheet.py", line 3, in <module>
    predictor = torch.hub.load("unilight/sheet:v0.2.5", "sheet_ssqa",
                ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/opt/conda/lib/python3.11/site-packages/torch/hub.py", line 647, in load
    model = _load_local(repo_or_dir, model, *args, **kwargs)
            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/opt/conda/lib/python3.11/site-packages/torch/hub.py", line 677, in _load_local
    hub_module = _import_module(MODULE_HUBCONF, hubconf_path)
                 ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/opt/conda/lib/python3.11/site-packages/torch/hub.py", line 115, in _import_module
    spec.loader.exec_module(module)
  File "<frozen importlib._bootstrap_external>", line 936, in exec_module
  File "<frozen importlib._bootstrap_external>", line 1074, in get_code
  File "<frozen importlib._bootstrap_external>", line 1004, in source_to_code
  File "<frozen importlib._bootstrap>", line 241, in _call_with_frames_removed
  File "/root/.cache/torch/hub/unilight_sheet_v0.2.5/hubconf.py", line 130
    cpu=True
        ^^^^
SyntaxError: invalid syntax. Perhaps you forgot a comma?

[Charles20021201]

+1

[unilight]

Hi @cindity , thanks for reporting this!

Unfortunately I am a bit occupied recently -- I might be able to fix this in February. I think the error is quite straightforward -- I am missing a comma. If you want to use the model now, please go ahead and fix the file in your local cache.

Sorry for this again!