improve training data path compatibility

josStorer · Jul 10, 2023 · 0edec68 · 0edec68
1 parent ee63dc2
commit 0edec68
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/frontend/src/pages/Train.tsx b/frontend/src/pages/Train.tsx
@@ -281,7 +281,9 @@ const LoraFinetune: FC = observer(() => {
  if (!ok)
  return;
 
- const convertedDataPath = `./finetune/json2binidx_tool/data/${dataParams.dataPath.split(/[\/\\]/).pop()!.split('.')[0]}_text_document`;
+ const convertedDataPath = './finetune/json2binidx_tool/data/' +
+ dataParams.dataPath.replace(/[\/\\]$/, '').split(/[\/\\]/).pop()!.split('.')[0] +
+ '_text_document';
  if (!await FileExists(convertedDataPath + '.idx')) {
  toast(t('Please convert data first.'), { type: 'error' });
  return;
@@ -432,7 +434,10 @@ const LoraFinetune: FC = observer(() => {
  return;
  const outputPrefix = './finetune/json2binidx_tool/data/' +
  dataParams.dataPath.replace(/[\/\\]$/, '').split(/[\/\\]/).pop()!.split('.')[0];
- ConvertData(commonStore.settings.customPythonPath, dataParams.dataPath, outputPrefix, dataParams.vocabPath).then(async () => {
+ ConvertData(commonStore.settings.customPythonPath,
+ dataParams.dataPath.replaceAll('\\', '/'),
+ outputPrefix,
+ dataParams.vocabPath).then(async () => {
  if (!await FileExists(outputPrefix + '_text_document.idx')) {
  toast(t('Failed to convert data') + ' - ' + await GetPyError(), { type: 'error' });
  } else {