lambda-audio-v6 / src /environment.py
KeisukeMiyamoto's picture
Add simple LambdaAudio v6 ZeroGPU transcription demo
82e11f0 verified
Raw History Blame Contribute Delete
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"]