Spaces:

cpu4dream
/

llava-small-open-elm-aimv2

Running

Camil Ziane commited on 8 days ago

Commit

ce9f495

•

1 Parent(s): a646ee4

fix model path

Files changed (1) hide show

TinyLLaVA_Factory/tinyllava/serve/app.py CHANGED Viewed

@@ -348,7 +348,7 @@ if __name__ == "__main__":
     logger.info(gr.__version__)
     args = parse_args()
     model_name = args.model_name
-    DEFAULT_MODEL_PATH = args.model_path
     model, tokenizer, image_processor, context_len = load_pretrained_model(
         args.model_path,
         load_4bit=args.load_4bit,

     logger.info(gr.__version__)
     args = parse_args()
     model_name = args.model_name
+    # DEFAULT_MODEL_PATH = args.model_path
     model, tokenizer, image_processor, context_len = load_pretrained_model(
         args.model_path,
         load_4bit=args.load_4bit,