can-gaa-hou commited on
Commit
d175231
·
1 Parent(s): d5db947
Files changed (2) hide show
  1. app.py +1 -1
  2. convert_model.py +1 -1
app.py CHANGED
@@ -223,7 +223,7 @@ def eval_model(image_file, model, tokenizer):
223
 
224
  if __name__ == "__main__":
225
  parser = argparse.ArgumentParser()
226
- parser.add_argument("--weight-dir", type=str, default="./")
227
  parser.add_argument("--image-file", type=str, required=True)
228
  args = parser.parse_args()
229
  model_dir = args.weight_dir
 
223
 
224
  if __name__ == "__main__":
225
  parser = argparse.ArgumentParser()
226
+ parser.add_argument("--weight-dir", type=str, default="./weight")
227
  parser.add_argument("--image-file", type=str, required=True)
228
  args = parser.parse_args()
229
  model_dir = args.weight_dir
convert_model.py CHANGED
@@ -218,7 +218,7 @@ class OvModelForCausalLMWithEmb(GenerationMixin):
218
  class OVGotOcrModel(GenerationMixin):
219
  def __init__(self, model_dir, device, ov_config=None, compression_configuration=None):
220
  model_dir = Path(model_dir)
221
- self.config = AutoConfig.from_pretrained(model_dir, trust_remote_code=True, local_files_only=True)
222
  self.generation_config = GenerationConfig.from_model_config(self.config)
223
  self.vision_tower_high = core.compile_model(model_dir / VISION_TOWER_HIGH_NAME, device, ov_config)
224
  self.mm_projector_vary = core.compile_model(model_dir / PROJECTOR_VARY_NAME, device, ov_config)
 
218
  class OVGotOcrModel(GenerationMixin):
219
  def __init__(self, model_dir, device, ov_config=None, compression_configuration=None):
220
  model_dir = Path(model_dir)
221
+ self.config = AutoConfig.from_pretrained(model_dir, trust_remote_code=True)
222
  self.generation_config = GenerationConfig.from_model_config(self.config)
223
  self.vision_tower_high = core.compile_model(model_dir / VISION_TOWER_HIGH_NAME, device, ov_config)
224
  self.mm_projector_vary = core.compile_model(model_dir / PROJECTOR_VARY_NAME, device, ov_config)