Upload folder using huggingface_hub

generation_utils.py

@@ -424,16 +424,13 @@ class DreamGenerationMixin:
             # this allows user-defined logits control of the intermediate steps
             logits = generation_logits_hook_func(i, x, logits)
             probs = F.softmax(logits, dim=-1)
-            
-            logger.warn(f"remask_schedule: {remask_schedule}")
-            
+                        
             #ВОТ ТУТ МАСКИНГ ДОБАВЛЯЕМ!
             keep_mask = torch.ones_like(x, device=self.device, dtype=torch.bool)
             if remask_schedule and remask_schedule == "tau":
                 tau = remask_tau[i]
                 token_confidence = probs.amax(-1)
                 keep_mask = token_confidence >= tau
-                logger.warning(f"total remask tokens: {len(x) - keep_mask.sum()}")
                 x = torch.where(keep_mask, x, mask_token_id)