Skip to content

Commit

Permalink
removed local path
Browse files Browse the repository at this point in the history
Signed-off-by: ftgreat <[email protected]>
  • Loading branch information
ftgreat committed Jun 29, 2023
1 parent b9d04e7 commit 495fdd6
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions examples/Aquila/Aquila-chat/generate_chat_lora.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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()
Expand Down

0 comments on commit 495fdd6

Please sign in to comment.