111

README.md

@@ -36,6 +36,7 @@ model = AutoModelForCausalLM.from_pretrained(
     trust_remote_code=True)
 tokenizer = AutoTokenizer.from_pretrained(
     'NaughtyDog97/FormalEnhencedGPS-9B',
+    use_fast=False,
     trust_remote_code=True)
 
 # text prompt
@@ -140,6 +141,8 @@ formalization_model = AutoModelForCausalLM.from_pretrained(
 
 formalization_tokenizer = AutoTokenizer.from_pretrained(
     'NaughtyDog97/GeoFormalizer',
+    use_fast=True,
+    padding_side="right",
     trust_remote_code=True)
 
 
@@ -150,6 +153,7 @@ reason_model = AutoModelForCausalLM.from_pretrained(
     trust_remote_code=True)
 reason_tokenizer = AutoTokenizer.from_pretrained(
     'NaughtyDog97/FormalEnhencedGPS-9B',
+    use_fase=False
     trust_remote_code=True)
 
 

modeling_fegeo_llama.py

@@ -2776,7 +2776,6 @@ class FEGeoLlamaForCausalLM(LlamaForCausalLM, FEGeoMetaForCausalLM):
         image_aspect_ratio = getattr(model_cfg, "image_aspect_ratio", None)
         new_images = []
         if image_aspect_ratio == 'pad':
-            print(f'using padding')
             for image in images:
                 image = self.crop(image)
                 image = self.expand2square(image, (255, 255, 255))