Spaces:
Running on Zero
Running on Zero
Cc commited on
Commit ·
735e026
1
Parent(s): e340a84
fixed:bug
Browse files
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 |
}
|