We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 1a018c4 commit 835b6b7Copy full SHA for 835b6b7
pypef/hybrid/hybrid_model.py
@@ -397,7 +397,7 @@ def train_llm(self):
397
#x_llm_ttest_b = get_batches(self.x_llm_ttest, batch_size=self.batch_size, dtype=int)
398
if self.llm_key == 'prosst':
399
y_llm_ttest = self.llm_inference_function(
400
- x_sequences=self.x_llm_ttest,
+ xs=self.x_llm_ttest,
401
model=self.llm_base_model,
402
input_ids=self.input_ids,
403
attention_mask=self.llm_attention_mask,
0 commit comments