EngineX-Iluvatar/enginex-bi_series-image-classification
16
0
Fork 0
Code Issues Pull Requests Actions Projects Releases 8 Wiki Activity

更新 transformers_server.py
All checks were successful
Docker Build and Push / docker (push) Successful in 2m24s
Details

Browse Source
This commit is contained in:
aiyueqi
2026-05-07 17:22:34 +08:00
parent 2e2e127738
commit baed0b5003
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
transformers_server.py
View File

@@ -79,7 +79,7 @@ def load_model():
processor_class = resolve_transformers_class(config["processer_class"])
model_class = resolve_transformers_class(config["model_class"])
torch_dtype = resolve_torch_dtype(config["torch_dtype"])
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
device = torch.device("cuda")
logger.info(
f"model config: model_class={config['model_class']}, "