111
Browse files
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 |
-
|
|
|
|
|
|
|
| 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 |
|