jinwei/chyoso-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修改模型地址

Browse Source
This commit is contained in:
William Jin 2024-09-25 21:44:44 +08:00
parent 970e70bd8e
commit e962345754
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
got-ocr.py
View File

@ -3,8 +3,8 @@ import torch
from modeling_GOT import GOTQwenForCausalLM
tokenizer = AutoTokenizer.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True)
model = GOTQwenForCausalLM.from_pretrained('ucaslcl/GOT-OCR2_0', low_cpu_mem_usage=True, device_map='cuda', use_safetensors=True,
tokenizer = AutoTokenizer.from_pretrained('ucaslcl/got_ocr2_0', trust_remote_code=True)
model = GOTQwenForCausalLM.from_pretrained('ucaslcl/got_ocr2_0', low_cpu_mem_usage=True, device_map='cuda', use_safetensors=True,
pad_token_id=151643).eval()
model.to(device='cuda', dtype=torch.bfloat16)