We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
PaLM2TextGenerator
1 parent 3867390 commit e4f13c3Copy full SHA for e4f13c3
bigframes/ml/llm.py
@@ -144,7 +144,7 @@ def _from_bq(
144
kwargs: dict = {}
145
last_fitting = model.training_runs[-1]["trainingOptions"]
146
147
- dummy_text_generator = cls()
+ dummy_text_generator = cls(session=session)
148
for bf_param, _ in dummy_text_generator.__dict__.items():
149
bqml_param = _BQML_PARAMS_MAPPING.get(bf_param)
150
if bqml_param in last_fitting:
0 commit comments