Commit
·
d175231
1
Parent(s):
d5db947
fix bug
Browse files- app.py +1 -1
- 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
|
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)
|