修复了merge pre_release分支后导致参数名不一致的bug

This commit is contained in:
三洋三洋
2024-04-17 14:47:58 +00:00
parent 891a9c310a
commit 5a9138026f
2 changed files with 2 additions and 2 deletions

View File

@@ -13,7 +13,7 @@ from models.globals import MAX_TOKEN_SIZE
def inference(
model: TexTeller,
tokenizer: RobertaTokenizerFast,
imgs_path: Union[List[str], List[np.ndarray]],
imgs: Union[List[str], List[np.ndarray]],
inf_mode: str = 'cpu',
num_beams: int = 1,
) -> List[str]:

View File

@@ -1,7 +1,7 @@
#!/usr/bin/env bash
set -exu
export CHECKPOINT_DIR="default"
export CHECKPOINT_DIR="/home/lhy/code/TexTeller/src/models/ocr_model/train/train_result/TexTellerv3/checkpoint-648000"
export TOKENIZER_DIR="default"
streamlit run web.py