gx

chenyc

2025-06-12 7b72ac13a83764a662159d4a49b7fffb90476ecb

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
import * as tf from '../../dist/tfjs.esm';
 
import { getModelUris } from '../common/getModelUris';
import { fetchJson } from './fetchJson';
 
export async function loadWeightMap(
  uri: string | undefined,
  defaultModelName: string,
): Promise<tf.NamedTensorMap> {
  const { manifestUri, modelBaseUri } = getModelUris(uri, defaultModelName);
  // @ts-ignore
  const manifest = await fetchJson<tf.io.WeightsManifestConfig>(manifestUri);
  // if (manifest['weightsManifest']) manifest = manifest['weightsManifest'];
  return tf['io'].loadWeights(manifest, modelBaseUri);
}