zezeze97 commited on
Commit
5f96533
·
1 Parent(s): 27df1a3
Files changed (1) hide show
  1. README.md +6 -1
README.md CHANGED
@@ -36,7 +36,9 @@ model = AutoModelForCausalLM.from_pretrained(
36
  trust_remote_code=True)
37
  tokenizer = AutoTokenizer.from_pretrained(
38
  'NaughtyDog97/FormalEnhencedGPS-34B',
39
- trust_remote_code=True)
 
 
40
 
41
  # text prompt
42
  img_path = 'sample/4927.png'
@@ -140,6 +142,8 @@ formalization_model = AutoModelForCausalLM.from_pretrained(
140
 
141
  formalization_tokenizer = AutoTokenizer.from_pretrained(
142
  'NaughtyDog97/GeoFormalizer',
 
 
143
  trust_remote_code=True)
144
 
145
 
@@ -150,6 +154,7 @@ reason_model = AutoModelForCausalLM.from_pretrained(
150
  trust_remote_code=True)
151
  reason_tokenizer = AutoTokenizer.from_pretrained(
152
  'NaughtyDog97/FormalEnhencedGPS-34B',
 
153
  trust_remote_code=True)
154
 
155
 
 
36
  trust_remote_code=True)
37
  tokenizer = AutoTokenizer.from_pretrained(
38
  'NaughtyDog97/FormalEnhencedGPS-34B',
39
+ use_fast=False,
40
+ trust_remote_code=True,
41
+ )
42
 
43
  # text prompt
44
  img_path = 'sample/4927.png'
 
142
 
143
  formalization_tokenizer = AutoTokenizer.from_pretrained(
144
  'NaughtyDog97/GeoFormalizer',
145
+ use_fast=True,
146
+ padding_side="right",
147
  trust_remote_code=True)
148
 
149
 
 
154
  trust_remote_code=True)
155
  reason_tokenizer = AutoTokenizer.from_pretrained(
156
  'NaughtyDog97/FormalEnhencedGPS-34B',
157
+ use_fast=False,
158
  trust_remote_code=True)
159
 
160