Skip to content

Commit e4f13c3

Browse files
authored
fix: use explicit session in PaLM2TextGenerator (#651)
This is to avoid running into conflict with the global session which may have different options (e.g. location, connection etc.) set.
1 parent 3867390 commit e4f13c3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

bigframes/ml/llm.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -144,7 +144,7 @@ def _from_bq(
144144
kwargs: dict = {}
145145
last_fitting = model.training_runs[-1]["trainingOptions"]
146146

147-
dummy_text_generator = cls()
147+
dummy_text_generator = cls(session=session)
148148
for bf_param, _ in dummy_text_generator.__dict__.items():
149149
bqml_param = _BQML_PARAMS_MAPPING.get(bf_param)
150150
if bqml_param in last_fitting:

0 commit comments

Comments
 (0)