VideoQuest / src /app /engines.ts
jbilcke-hf's picture
jbilcke-hf HF staff
working on the menu
bf8d4d8
raw
history blame
683 Bytes
export type EngineType = "image" | "video"
export interface Engine {
type: EngineType
label: string
modelName: string
modelUrl: string
}
export const engines: Record<string, Engine> = {
image: {
type: "image",
label: "Image",
modelName: "SDXL 1.0",
modelUrl: "https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0",
},
video: {
type: "video",
label: "Video",
modelName: "Zeroscope V2 576w",
modelUrl: "https://huggingface.co/cerspense/zeroscope_v2_576w",
}
}
export const defaultEngine: EngineType = "image"
export const getEngine = (type?: EngineType): Engine => engines[type || defaultEngine] || engines[defaultEngine]