llm/SIA
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Renamed hf-api-token to api-token

Browse Source
This commit is contained in:
geens
2024-11-11 13:25:39 +01:00
parent 0d98a42df6
commit babc1c44b9
2 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
sia/config.py
View File

@@ -64,10 +64,10 @@ class Config:
help='LLM engine (default: local, env: SIA_LLM_ENGINE)'
)
parser.add_argument(
'--hf-api-token',
'--api-token',
type=str,
default=os.getenv('SIA_HF_API_TOKEN'),
help='Hugging Face access token (env: SIA_HF_API_TOKEN)'
default=os.getenv('SIA_API_TOKEN'),
help='API access token (env: SIA_API_TOKEN)'
)
parser.add_argument(
'--model',
@@ -127,9 +127,9 @@ class Config:
return self.args.llm_engine
@property
def hf_api_token(self) -> Optional[str]:
"""Hugging Face access token."""
return self.args.hf_api_token
def api_token(self) -> Optional[str]:
"""API access token."""
return self.args.api_token
@property
def model(self) -> str: