Update generate_model.py

generate_model.py

@@ -594,7 +594,7 @@ def generate(
     # print('model load over')
     config = model.config
     if tokenizer is None:
-        tokenizer = AutoTokenizer.from_pretrained(model, use_fast=False, model_max_length = config.tokenizer_model_max_length,
+        tokenizer = AutoTokenizer.from_pretrained(checkpoint_path, use_fast=False, model_max_length = config.tokenizer_model_max_length,
                 padding_side = config.tokenizer_padding_side)
     image_processor = model.vision_tower._image_processor
     context_len = getattr(config, 'max_sequence_length', 2048)
@@ -653,28 +653,6 @@ def generate(
 def tinyllava_phi_generate_parser():
     """Argument Parser"""
 
-    class KwargsParser(argparse.Action):
-        """Parser action class to parse kwargs of form key=value"""
-        def __call__(self, parser, namespace, values, option_string=None):
-            setattr(namespace, self.dest, dict())
-            for val in values:
-                if '=' not in val:
-                    raise ValueError(
-                        (
-                            'Argument parsing error, kwargs are expected in'
-                            ' the form of key=value.'
-                        )
-                    )
-                kwarg_k, kwarg_v = val.split('=')
-                try:
-                    converted_v = int(kwarg_v)
-                except ValueError:
-                    try:
-                        converted_v = float(kwarg_v)
-                    except ValueError:
-                        converted_v = kwarg_v            
-                getattr(namespace, self.dest)[kwarg_k] = converted_v
-
     parser = argparse.ArgumentParser('TinyLLaVA-Phi Generate Module')
     parser.add_argument(
         '--model',