Upload folder using huggingface_hub

README.md

@@ -52,4 +52,6 @@ generations = [
 ]
 
 print(generations[0].split(tokenizer.eos_token)[0])
-```
+```
+
+We adapt the original Dream-7B-Coder-Instruct with soft-masking (SM). More information on SM will be linked here in the future.

generation_utils.py

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