7f2b5c41126bea4704e279deea0068f0462d5b49,python/baseline/services.py,LanguageModelService,load,#Any#Any#,245

Before Change

        kwargs["batchsz"] = 1
        vocabs = load_vocabs(directory)

        vectorizers = load_vectorizers(directory)
        model_basename = find_model_basename(directory)
        be = kwargs.get("backend", "tf")
        import_user_module("baseline.{}.embeddings".format(be))
        import_user_module("baseline.{}.lm".format(be))
        model = load_lang_model(model_basename, **kwargs)
        return cls(vocabs, vectorizers, model)

    // Do a greedy decode for now, everything else will be super slow
    def predict(self, tokens, **kwargs):
        mxlen = kwargs.get("mxlen", 10)

After Change

    @classmethod
    def load(cls, bundle, **kwargs):
        kwargs["batchsz"] = 1
        return super(LanguageModelService, cls).load(bundle, **kwargs)

    // Do a greedy decode for now, everything else will be super slow
    def predict(self, tokens, **kwargs):
        mxlen = kwargs.get("mxlen", 10)

Instances