[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
704e84c055
commit
8de6ae759d
|
@ -79,7 +79,7 @@ class PI0FASTConfig(PreTrainedConfig):
|
|||
precision: str = "bfloat16"
|
||||
grad_clip_norm: float = 1
|
||||
|
||||
# Allows padding/truncation of generated action tokens during detokenization to ensure decoding.
|
||||
# Allows padding/truncation of generated action tokens during detokenization to ensure decoding.
|
||||
# In the original version, tensors of 0s were generated if shapes didn't match for stable decoding.
|
||||
relaxed_decoding: bool = True
|
||||
|
||||
|
|
Loading…
Reference in New Issue