Upload modeling_qed.py with huggingface_hub
Browse files- 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 =
|
| 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 |
"""
|