smoorsmith
/

Dream-coder-test

Text Generation

feature-extraction

Model card Files Files and versions

smoorsmith commited on Sep 24

Commit

b0e266c

·

verified ·

1 Parent(s): 8246c6d

Update generation_utils.py

Files changed (1) hide show

generation_utils.py +1 -1

generation_utils.py CHANGED Viewed

@@ -364,7 +364,7 @@ class DreamGenerationMixin:
         )
         #############################################
-        print(f"SM parameters used in generation: {sm_args}")
         result = self._sample(
             input_ids,

         )
         #############################################
+        print(f"SM parameters used in generation: {sm_args}", flush=True)
         result = self._sample(
             input_ids,