Skip to content

Commit 835b6b7

Browse files
committed
x_sequences -> xs
1 parent 1a018c4 commit 835b6b7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pypef/hybrid/hybrid_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -397,7 +397,7 @@ def train_llm(self):
397397
#x_llm_ttest_b = get_batches(self.x_llm_ttest, batch_size=self.batch_size, dtype=int)
398398
if self.llm_key == 'prosst':
399399
y_llm_ttest = self.llm_inference_function(
400-
x_sequences=self.x_llm_ttest,
400+
xs=self.x_llm_ttest,
401401
model=self.llm_base_model,
402402
input_ids=self.input_ids,
403403
attention_mask=self.llm_attention_mask,

0 commit comments

Comments
 (0)