levossadtchi commited on
Commit
695b8a4
·
verified ·
1 Parent(s): ed260ca

Upload modeling_qed.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. modeling_qed.py +1 -1
modeling_qed.py CHANGED
@@ -249,7 +249,7 @@ class QEDForCausalLM(PreTrainedModel, GenerationMixin):
249
  temperature: float = 0.8,
250
  top_k: int | None = 50,
251
  eos_token_id: Optional[int] = None,
252
- do_sample: bool = False,
253
  **kwargs,
254
  ) -> torch.LongTensor:
255
  """
 
249
  temperature: float = 0.8,
250
  top_k: int | None = 50,
251
  eos_token_id: Optional[int] = None,
252
+ do_sample: bool = True,
253
  **kwargs,
254
  ) -> torch.LongTensor:
255
  """