loadWeightMap.js 878 B

1234567891011121314151617181920
  1. import { __awaiter, __generator } from "tslib";
  2. import * as tf from '@tensorflow/tfjs-core';
  3. import { getModelUris } from '../common/getModelUris';
  4. import { fetchJson } from './fetchJson';
  5. export function loadWeightMap(uri, defaultModelName) {
  6. return __awaiter(this, void 0, void 0, function () {
  7. var _a, manifestUri, modelBaseUri, manifest;
  8. return __generator(this, function (_b) {
  9. switch (_b.label) {
  10. case 0:
  11. _a = getModelUris(uri, defaultModelName), manifestUri = _a.manifestUri, modelBaseUri = _a.modelBaseUri;
  12. return [4 /*yield*/, fetchJson(manifestUri)];
  13. case 1:
  14. manifest = _b.sent();
  15. return [2 /*return*/, tf.io.loadWeights(manifest, modelBaseUri)];
  16. }
  17. });
  18. });
  19. }
  20. //# sourceMappingURL=loadWeightMap.js.map