TuKoResearch
/

librilightGPT

AuriStream.AuriStream

Model card Files Files and versions

klemenk commited on May 21, 2025

Commit

f163227

·

verified ·

1 Parent(s): 980097b

Update modeling_auristream.py

Files changed (1) hide show

modeling_auristream.py +1 -1

modeling_auristream.py CHANGED Viewed

@@ -149,7 +149,7 @@ class AuriStream(PreTrainedModel):
         if output_logits and return_dict:
             model_output = CausalLMOutput(
-                logits=all_logits,
             )
             return model_output

         if output_logits and return_dict:
             model_output = CausalLMOutput(
+                logits=logits,
             )
             return model_output