|
| 1 | +const makeModelConstants = (name: string) => ({ |
| 2 | + modelSource: `https://huggingface.co/mock/${name}/model.pte`, |
| 3 | + tokenizerSource: `https://huggingface.co/mock/${name}/tokenizer.json`, |
| 4 | + tokenizerConfigSource: `https://huggingface.co/mock/${name}/tokenizer_config.json`, |
| 5 | +}); |
| 6 | + |
| 7 | +export const QWEN3_0_6B_QUANTIZED = makeModelConstants('qwen3-0.6b-quantized'); |
| 8 | +export const QWEN3_0_6B = makeModelConstants('qwen3-0.6b'); |
| 9 | +export const QWEN3_1_7B = makeModelConstants('qwen3-1.7b'); |
| 10 | +export const QWEN3_1_7B_QUANTIZED = makeModelConstants('qwen3-1.7b-quantized'); |
| 11 | +export const QWEN3_4B = makeModelConstants('qwen3-4b'); |
| 12 | +export const QWEN3_4B_QUANTIZED = makeModelConstants('qwen3-4b-quantized'); |
| 13 | +export const LLAMA3_2_1B = makeModelConstants('llama3.2-1b'); |
| 14 | +export const LLAMA3_2_1B_QLORA = makeModelConstants('llama3.2-1b-qlora'); |
| 15 | +export const LLAMA3_2_1B_SPINQUANT = makeModelConstants('llama3.2-1b-spinquant'); |
| 16 | +export const LLAMA3_2_3B = makeModelConstants('llama3.2-3b'); |
| 17 | +export const LLAMA3_2_3B_QLORA = makeModelConstants('llama3.2-3b-qlora'); |
| 18 | +export const LLAMA3_2_3B_SPINQUANT = makeModelConstants('llama3.2-3b-spinquant'); |
| 19 | +export const HAMMER2_1_0_5B = makeModelConstants('hammer2.1-0.5b'); |
| 20 | +export const HAMMER2_1_0_5B_QUANTIZED = makeModelConstants('hammer2.1-0.5b-quantized'); |
| 21 | +export const HAMMER2_1_1_5B = makeModelConstants('hammer2.1-1.5b'); |
| 22 | +export const HAMMER2_1_1_5B_QUANTIZED = makeModelConstants('hammer2.1-1.5b-quantized'); |
| 23 | +export const HAMMER2_1_3B = makeModelConstants('hammer2.1-3b'); |
| 24 | +export const HAMMER2_1_3B_QUANTIZED = makeModelConstants('hammer2.1-3b-quantized'); |
| 25 | +export const QWEN2_5_0_5B = makeModelConstants('qwen2.5-0.5b'); |
| 26 | +export const QWEN2_5_0_5B_QUANTIZED = makeModelConstants('qwen2.5-0.5b-quantized'); |
| 27 | +export const QWEN2_5_1_5B = makeModelConstants('qwen2.5-1.5b'); |
| 28 | +export const QWEN2_5_1_5B_QUANTIZED = makeModelConstants('qwen2.5-1.5b-quantized'); |
| 29 | +export const QWEN2_5_3B = makeModelConstants('qwen2.5-3b'); |
| 30 | +export const QWEN2_5_3B_QUANTIZED = makeModelConstants('qwen2.5-3b-quantized'); |
| 31 | +export const PHI_4_MINI_4B = makeModelConstants('phi4-mini-4b'); |
| 32 | +export const PHI_4_MINI_4B_QUANTIZED = makeModelConstants('phi4-mini-4b-quantized'); |
| 33 | +export const LFM2_5_1_2B_INSTRUCT = makeModelConstants('lfm2.5-1.2b-instruct'); |
| 34 | +export const LFM2_5_1_2B_INSTRUCT_QUANTIZED = makeModelConstants('lfm2.5-1.2b-instruct-quantized'); |
| 35 | +export const WHISPER_TINY_EN = 'whisper-tiny-en'; |
| 36 | + |
| 37 | +export const LLMModule = { |
| 38 | + fromCustomModel: jest.fn(), |
| 39 | +}; |
| 40 | + |
| 41 | +export const SpeechToTextModule = { |
| 42 | + fromModelName: jest.fn(), |
| 43 | +}; |
| 44 | + |
| 45 | +export const TextEmbeddingsModule = { |
| 46 | + fromCustomModel: jest.fn(), |
| 47 | +}; |
| 48 | + |
| 49 | +export const ResourceFetcher = { |
| 50 | + fetch: jest.fn(), |
| 51 | +}; |
0 commit comments