From 495fdd63d8528ea1f2db144d118d52b201e0c47d Mon Sep 17 00:00:00 2001 From: ftgreat Date: Thu, 29 Jun 2023 12:27:09 +0800 Subject: [PATCH] removed local path Signed-off-by: ftgreat --- examples/Aquila/Aquila-chat/generate_chat_lora.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/examples/Aquila/Aquila-chat/generate_chat_lora.py b/examples/Aquila/Aquila-chat/generate_chat_lora.py index cd4affd5..dddd0536 100755 --- a/examples/Aquila/Aquila-chat/generate_chat_lora.py +++ b/examples/Aquila/Aquila-chat/generate_chat_lora.py @@ -3,7 +3,6 @@ # Licensed under the Apache License, Version 2.0 (the "License") import os import torch -import sys;sys.path.append("/mnt/yzd/git/FlagAI/") from flagai.auto_model.auto_loader import AutoLoader from flagai.model.predictor.predictor import Predictor from flagai.data.tokenizer import Tokenizer @@ -21,7 +20,7 @@ use_cache=True, fp16=True, device='cuda', - adapter_dir='/mnt/yzd/git/FlagAI/examples/Aquila/Aquila-chat/checkpoints_out/aquila_experiment/2023062909') + adapter_dir='directory of adapter files') # eg: /mnt/yzd/git/FlagAI/examples/Aquila/Aquila-chat/checkpoints_out/aquila_experiment/2023062909 model = loader.get_model() tokenizer = loader.get_tokenizer()