aws-cdk-neuronx-patterns
Version:
> [!WARNING] > This library is experimental module.
20 lines • 3.69 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
const model_1 = require("./model");
describe("Model.fromHuggingFace", () => {
it("When the number of parameters is included in the model ID, the number of parameters should be inferred", () => {
const model = model_1.Model.fromHuggingFace("example/example-7b-chat");
expect(model.options.parameters).toStrictEqual(model_1.Parameters.billion(7));
});
it("When the number of parameters is not included in the model ID, an error should be thrown", () => {
const when = () => model_1.Model.fromHuggingFace("example/example-chat");
expect(when).toThrow("The number of parameters cannot be inferred from the model ID. Set optional parameters.");
});
it("When the number of parameters is set as an option, it should be set as is", () => {
const model = model_1.Model.fromHuggingFace("example/no1b-chat", {
parameters: model_1.Parameters.billion(7),
});
expect(model.options.parameters).toStrictEqual(model_1.Parameters.billion(7));
});
});
//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoibW9kZWwudGVzdC5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uLy4uL3NyYy9iYXNlL25ldXJvbngvbW9kZWwudGVzdC50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiOztBQUFBLG1DQUE0QztBQUU1QyxRQUFRLENBQUMsdUJBQXVCLEVBQUUsR0FBRyxFQUFFO0lBQ3JDLEVBQUUsQ0FBQyx3R0FBd0csRUFBRSxHQUFHLEVBQUU7UUFDaEgsTUFBTSxLQUFLLEdBQUcsYUFBSyxDQUFDLGVBQWUsQ0FBQyx5QkFBeUIsQ0FBQyxDQUFDO1FBQy9ELE1BQU0sQ0FBQyxLQUFLLENBQUMsT0FBTyxDQUFDLFVBQVUsQ0FBQyxDQUFDLGFBQWEsQ0FBQyxrQkFBVSxDQUFDLE9BQU8sQ0FBQyxDQUFDLENBQUMsQ0FBQyxDQUFDO0lBQ3hFLENBQUMsQ0FBQyxDQUFDO0lBQ0gsRUFBRSxDQUFDLDBGQUEwRixFQUFFLEdBQUcsRUFBRTtRQUNsRyxNQUFNLElBQUksR0FBRyxHQUFHLEVBQUUsQ0FBQyxhQUFLLENBQUMsZUFBZSxDQUFDLHNCQUFzQixDQUFDLENBQUM7UUFDakUsTUFBTSxDQUFDLElBQUksQ0FBQyxDQUFDLE9BQU8sQ0FDbEIseUZBQXlGLENBQzFGLENBQUM7SUFDSixDQUFDLENBQUMsQ0FBQztJQUNILEVBQUUsQ0FBQywyRUFBMkUsRUFBRSxHQUFHLEVBQUU7UUFDbkYsTUFBTSxLQUFLLEdBQUcsYUFBSyxDQUFDLGVBQWUsQ0FBQyxtQkFBbUIsRUFBRTtZQUN2RCxVQUFVLEVBQUUsa0JBQVUsQ0FBQyxPQUFPLENBQUMsQ0FBQyxDQUFDO1NBQ2xDLENBQUMsQ0FBQztRQUNILE1BQU0sQ0FBQyxLQUFLLENBQUMsT0FBTyxDQUFDLFVBQVUsQ0FBQyxDQUFDLGFBQWEsQ0FBQyxrQkFBVSxDQUFDLE9BQU8sQ0FBQyxDQUFDLENBQUMsQ0FBQyxDQUFDO0lBQ3hFLENBQUMsQ0FBQyxDQUFDO0FBQ0wsQ0FBQyxDQUFDLENBQUMiLCJzb3VyY2VzQ29udGVudCI6WyJpbXBvcnQgeyBNb2RlbCwgUGFyYW1ldGVycyB9IGZyb20gXCIuL21vZGVsXCI7XG5cbmRlc2NyaWJlKFwiTW9kZWwuZnJvbUh1Z2dpbmdGYWNlXCIsICgpID0+IHtcbiAgaXQoXCJXaGVuIHRoZSBudW1iZXIgb2YgcGFyYW1ldGVycyBpcyBpbmNsdWRlZCBpbiB0aGUgbW9kZWwgSUQsIHRoZSBudW1iZXIgb2YgcGFyYW1ldGVycyBzaG91bGQgYmUgaW5mZXJyZWRcIiwgKCkgPT4ge1xuICAgIGNvbnN0IG1vZGVsID0gTW9kZWwuZnJvbUh1Z2dpbmdGYWNlKFwiZXhhbXBsZS9leGFtcGxlLTdiLWNoYXRcIik7XG4gICAgZXhwZWN0KG1vZGVsLm9wdGlvbnMucGFyYW1ldGVycykudG9TdHJpY3RFcXVhbChQYXJhbWV0ZXJzLmJpbGxpb24oNykpO1xuICB9KTtcbiAgaXQoXCJXaGVuIHRoZSBudW1iZXIgb2YgcGFyYW1ldGVycyBpcyBub3QgaW5jbHVkZWQgaW4gdGhlIG1vZGVsIElELCBhbiBlcnJvciBzaG91bGQgYmUgdGhyb3duXCIsICgpID0+IHtcbiAgICBjb25zdCB3aGVuID0gKCkgPT4gTW9kZWwuZnJvbUh1Z2dpbmdGYWNlKFwiZXhhbXBsZS9leGFtcGxlLWNoYXRcIik7XG4gICAgZXhwZWN0KHdoZW4pLnRvVGhyb3coXG4gICAgICBcIlRoZSBudW1iZXIgb2YgcGFyYW1ldGVycyBjYW5ub3QgYmUgaW5mZXJyZWQgZnJvbSB0aGUgbW9kZWwgSUQuIFNldCBvcHRpb25hbCBwYXJhbWV0ZXJzLlwiLFxuICAgICk7XG4gIH0pO1xuICBpdChcIldoZW4gdGhlIG51bWJlciBvZiBwYXJhbWV0ZXJzIGlzIHNldCBhcyBhbiBvcHRpb24sIGl0IHNob3VsZCBiZSBzZXQgYXMgaXNcIiwgKCkgPT4ge1xuICAgIGNvbnN0IG1vZGVsID0gTW9kZWwuZnJvbUh1Z2dpbmdGYWNlKFwiZXhhbXBsZS9ubzFiLWNoYXRcIiwge1xuICAgICAgcGFyYW1ldGVyczogUGFyYW1ldGVycy5iaWxsaW9uKDcpLFxuICAgIH0pO1xuICAgIGV4cGVjdChtb2RlbC5vcHRpb25zLnBhcmFtZXRlcnMpLnRvU3RyaWN0RXF1YWwoUGFyYW1ldGVycy5iaWxsaW9uKDcpKTtcbiAgfSk7XG59KTtcbiJdfQ==