Add support for MiniMax H3 T2V and I2V training
This commit is contained in:
@@ -1,6 +1,6 @@
|
||||
import { NextResponse } from 'next/server';
|
||||
import prisma from '@/server/prisma';
|
||||
import { defaultTrainFolder, defaultDatasetsFolder } from '@/paths';
|
||||
import { defaultTrainFolder, defaultDatasetsFolder, defaultModelsFolder } from '@/paths';
|
||||
import { flushCache } from '@/server/settings';
|
||||
|
||||
export async function GET() {
|
||||
@@ -18,6 +18,13 @@ export async function GET() {
|
||||
if (!settingsObject.DATASETS_FOLDER || settingsObject.DATASETS_FOLDER === '') {
|
||||
settingsObject.DATASETS_FOLDER = defaultDatasetsFolder;
|
||||
}
|
||||
// MODELS_PATH from the env file always takes precedence over the setting
|
||||
if (process.env.MODELS_PATH && process.env.MODELS_PATH.trim() !== '') {
|
||||
settingsObject.MODELS_PATH = process.env.MODELS_PATH;
|
||||
} else if (!settingsObject.MODELS_PATH || settingsObject.MODELS_PATH === '') {
|
||||
// if MODELS_PATH is not set, use default
|
||||
settingsObject.MODELS_PATH = defaultModelsFolder;
|
||||
}
|
||||
return NextResponse.json(settingsObject);
|
||||
} catch (error) {
|
||||
return NextResponse.json({ error: 'Failed to fetch settings' }, { status: 500 });
|
||||
@@ -27,7 +34,7 @@ export async function GET() {
|
||||
export async function POST(request: Request) {
|
||||
try {
|
||||
const body = await request.json();
|
||||
const { HF_TOKEN, TRAINING_FOLDER, DATASETS_FOLDER } = body;
|
||||
const { HF_TOKEN, TRAINING_FOLDER, DATASETS_FOLDER, MODELS_PATH } = body;
|
||||
|
||||
// Upsert both settings
|
||||
await Promise.all([
|
||||
@@ -46,6 +53,11 @@ export async function POST(request: Request) {
|
||||
update: { value: DATASETS_FOLDER },
|
||||
create: { key: 'DATASETS_FOLDER', value: DATASETS_FOLDER },
|
||||
}),
|
||||
prisma.settings.upsert({
|
||||
where: { key: 'MODELS_PATH' },
|
||||
update: { value: MODELS_PATH },
|
||||
create: { key: 'MODELS_PATH', value: MODELS_PATH },
|
||||
}),
|
||||
]);
|
||||
|
||||
flushCache();
|
||||
|
||||
Reference in New Issue
Block a user