theroyallab/tabbyAPI
1
0
Fork 0
mirror of https://github.com/theroyallab/tabbyAPI.git synced 2026-03-15 00:07:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Transformers Utils: Fix file read

Browse Source 
Use asynchronous JSON reading

Signed-off-by: kingbri <bdashore3@proton.me>
This commit is contained in:
kingbri
2024-09-10 22:41:39 -04:00
parent 62beb2b1c8
commit 7baef05b49
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
common/transformers_utils.py
View File

@@ -23,7 +23,8 @@ class GenerationConfig(BaseModel):
async with aiofiles.open( async with aiofiles.open(
generation_config_path, "r", encoding="utf8" generation_config_path, "r", encoding="utf8"
) as generation_config_json: ) as generation_config_json:
generation_config_dict = json.load(generation_config_json) contents = await generation_config_json.read()
generation_config_dict = json.loads(contents)
return cls.model_validate(generation_config_dict) return cls.model_validate(generation_config_dict)
def eos_tokens(self): def eos_tokens(self):