Cc commited on
Commit
735e026
·
1 Parent(s): e340a84

fixed:bug

Browse files
Files changed (1) hide show
  1. longstream/demo/backend.py +2 -2
longstream/demo/backend.py CHANGED
@@ -188,8 +188,8 @@ def _resolve_demo_checkpoint(checkpoint: str) -> str:
188
  return os.path.abspath(candidate)
189
 
190
  hf_cfg = {
191
- "repo_id": os.getenv("LONGSTREAM_HF_REPO"),
192
- "filename": os.getenv("LONGSTREAM_HF_FILE"),
193
  "revision": os.getenv("LONGSTREAM_HF_REVISION"),
194
  "local_dir": os.getenv("LONGSTREAM_HF_LOCAL_DIR", "checkpoints"),
195
  }
 
188
  return os.path.abspath(candidate)
189
 
190
  hf_cfg = {
191
+ "repo_id": os.getenv("LONGSTREAM_HF_REPO", "NicolasCC/LongStream"),
192
+ "filename": os.getenv("LONGSTREAM_HF_FILE", "50_longstream.pt"),
193
  "revision": os.getenv("LONGSTREAM_HF_REVISION"),
194
  "local_dir": os.getenv("LONGSTREAM_HF_LOCAL_DIR", "checkpoints"),
195
  }