perf: 微调截止词

This commit is contained in:
archer 2023-03-26 15:27:57 +08:00
parent 9125910cfe
commit e547893ac1
No known key found for this signature in database
GPG Key ID: 166CA6BF2383B2BB
4 changed files with 9 additions and 8 deletions

View File

@ -91,8 +91,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
messages: formatPrompts, messages: formatPrompts,
frequency_penalty: 0.5, // 越大,重复内容越少 frequency_penalty: 0.5, // 越大,重复内容越少
presence_penalty: -0.5, // 越大,越容易出现新内容 presence_penalty: -0.5, // 越大,越容易出现新内容
stream: true, stream: true
stop: ['。!?.!.']
}, },
{ {
timeout: 40000, timeout: 40000,

View File

@ -78,7 +78,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
// model.trainingTimes > 0 ? modelConstantsData.trainedMaxToken : modelConstantsData.maxToken, // model.trainingTimes > 0 ? modelConstantsData.trainedMaxToken : modelConstantsData.maxToken,
// presence_penalty: -0.5, // 越大,越容易出现新内容 // presence_penalty: -0.5, // 越大,越容易出现新内容
// frequency_penalty: 0.5, // 越大,重复内容越少 // frequency_penalty: 0.5, // 越大,重复内容越少
// stop: [`</s>`, '。!?.!.'] // stop: [`###`]
// }); // });
// 发出请求 // 发出请求
const chatResponse = await chatAPI.createCompletion( const chatResponse = await chatAPI.createCompletion(
@ -93,7 +93,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
: modelConstantsData.maxToken, : modelConstantsData.maxToken,
presence_penalty: -0.5, // 越大,越容易出现新内容 presence_penalty: -0.5, // 越大,越容易出现新内容
frequency_penalty: 0.5, // 越大,重复内容越少 frequency_penalty: 0.5, // 越大,重复内容越少
stop: [`</s>`, '。!?.!.'] stop: [`###`, '。!?.!.']
}, },
{ {
timeout: 40000, timeout: 40000,

View File

@ -47,7 +47,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
// 获取 openai 的训练情况 // 获取 openai 的训练情况
const { data } = await openai.retrieveFineTune(training.tuneId, { httpsAgent }); const { data } = await openai.retrieveFineTune(training.tuneId, { httpsAgent });
// console.log(data);
if (data.status === OpenAiTuneStatusEnum.succeeded) { if (data.status === OpenAiTuneStatusEnum.succeeded) {
// 删除训练文件 // 删除训练文件
openai.deleteFile(data.training_files[0].id, { httpsAgent }); openai.deleteFile(data.training_files[0].id, { httpsAgent });

View File

@ -28,6 +28,8 @@ import { useRouter } from 'next/router';
import { useConfirm } from '@/hooks/useConfirm'; import { useConfirm } from '@/hooks/useConfirm';
import { useRequest } from '@/hooks/useRequest'; import { useRequest } from '@/hooks/useRequest';
import { DataItemSchema } from '@/types/mongoSchema'; import { DataItemSchema } from '@/types/mongoSchema';
import { customAlphabet } from 'nanoid';
const nanoid = customAlphabet('.,', 1);
const CreateDataModal = dynamic(() => import('./components/CreateDataModal')); const CreateDataModal = dynamic(() => import('./components/CreateDataModal'));
const ImportDataModal = dynamic(() => import('./components/ImportDataModal')); const ImportDataModal = dynamic(() => import('./components/ImportDataModal'));
@ -83,8 +85,8 @@ const DataList = () => {
// 生成 jsonl // 生成 jsonl
data.forEach((item) => { data.forEach((item) => {
const result = JSON.stringify({ const result = JSON.stringify({
prompt: `${item.q.toLocaleLowerCase()}</s>`, prompt: `${item.q.toLocaleLowerCase()}${nanoid()}</s>`,
completion: ` ${item.a}</s>` completion: ` ${item.a}###`
}); });
text += `${result}\n`; text += `${result}\n`;
}); });
@ -181,7 +183,7 @@ const DataList = () => {
</Button> </Button>
<Menu> <Menu>
<MenuButton as={Button} mr={2} size={'sm'}> <MenuButton as={Button} mr={2} size={'sm'} isLoading={isExporting}>
</MenuButton> </MenuButton>
<MenuList> <MenuList>