Spaces:
Sleeping
Sleeping
Download src/environment.py from KeisukeMiyamoto/lambda-audio-v6: direct link, hf CLI and curl.
- Browser
- Download file 406 Bytes
-
https://huggingface.co/spaces/KeisukeMiyamoto/lambda-audio-v6/resolve/main/src/environment.py
- Command line
-
hf download hf://spaces/KeisukeMiyamoto/lambda-audio-v6/src/environment.py
-
curl -L -o environment.py https://huggingface.co/spaces/KeisukeMiyamoto/lambda-audio-v6/resolve/main/src/environment.py
406 Bytes
| import os | |
| from pathlib import Path | |
| from dotenv import load_dotenv | |
| CUDA_ALLOCATOR_CONFIGURATION = "expandable_segments:True,garbage_collection_threshold:0.8" | |
| def configure_cuda_allocator() -> None: | |
| os.environ["PYTORCH_ALLOC_CONF"] = CUDA_ALLOCATOR_CONFIGURATION | |
| def load_hf_token(dotenv_path: Path = Path(".env")) -> str: | |
| load_dotenv(dotenv_path=dotenv_path) | |
| return os.environ["HF_TOKEN"] | |