This commit is contained in:
Re-bin
2025-02-25 22:50:03 +08:00
parent 69829b0738
commit 9e8eb94591

View File

@@ -8,9 +8,6 @@ import sys
warnings.filterwarnings("ignore") warnings.filterwarnings("ignore")
logging.getLogger("httpx").setLevel(logging.WARNING) logging.getLogger("httpx").setLevel(logging.WARNING)
# Add the parent directory of 'videorag' to sys.path
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
import argparse import argparse
parser = argparse.ArgumentParser(description="Set sub-category and CUDA device.") parser = argparse.ArgumentParser(description="Set sub-category and CUDA device.")
parser.add_argument('--collection', type=str, default='4-rag-lecture') parser.add_argument('--collection', type=str, default='4-rag-lecture')