Skip to content

GitLab

Unverified Commit 29006e39 authored by Tong Gao's avatar Tong Gao Committed by GitHub
Browse files

[Fix] Fix circular import of PromptTemplate (#71)

parent 1e445417
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
opencompass/openicl/icl_retriever/icl_base_retriever.py
View file @ 29006e39
...@@ -4,7 +4,7 @@ from typing import Dict, List, Optional ...@@ -4,7 +4,7 @@ from typing import Dict, List, Optional
from mmengine.dist import is_main_process from mmengine.dist import is_main_process
from opencompass.openicl import PromptTemplate from opencompass.openicl.icl_prompt_template import PromptTemplate
from opencompass.utils.prompt import PromptList from opencompass.utils.prompt import PromptList
......
opencompass/openicl/icl_retriever/icl_mdl_retriever.py
View file @ 29006e39
...@@ -7,7 +7,7 @@ import torch ...@@ -7,7 +7,7 @@ import torch
import tqdm import tqdm
from transformers import AutoModelForCausalLM from transformers import AutoModelForCausalLM
from opencompass.openicl import PromptTemplate from opencompass.openicl.icl_prompt_template import PromptTemplate
from opencompass.openicl.icl_retriever.icl_topk_retriever import TopkRetriever from opencompass.openicl.icl_retriever.icl_topk_retriever import TopkRetriever
from opencompass.openicl.utils.logging import get_logger from opencompass.openicl.utils.logging import get_logger
from opencompass.registry import ICL_PROMPT_TEMPLATES, ICL_RETRIEVERS from opencompass.registry import ICL_PROMPT_TEMPLATES, ICL_RETRIEVERS
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment