ai-toolkit / ui /src /paths.ts
jbilcke-hf's picture
Convert AI-Toolkit to a HF Space
8822914
import path from 'path';
// our fork is designed to be deployed as a Hugging Face space,
// and people will want to train long running jobs so persistent storage should be the default
// export const TOOLKIT_ROOT = path.resolve('@', '..', '..');
export const TOOLKIT_ROOT = '/data/ai-toolkit/';
export const defaultTrainFolder = path.join(TOOLKIT_ROOT, 'output');
export const defaultDatasetsFolder = path.join(TOOLKIT_ROOT, 'datasets');
export const defaultDataRoot = path.join(TOOLKIT_ROOT, 'data');