Unverified Commit e769ca3d authored by Pranith Pashikanti's avatar Pranith Pashikanti Committed by GitHub
Browse files

Added paper links in logitprocess.py (#25482)

parent 5c67682b
...@@ -1333,6 +1333,8 @@ class WhisperTimeStampLogitsProcessor(LogitsProcessor): ...@@ -1333,6 +1333,8 @@ class WhisperTimeStampLogitsProcessor(LogitsProcessor):
Whisper specific Processor. This processor can be used to force a list of tokens. The processor will set their log Whisper specific Processor. This processor can be used to force a list of tokens. The processor will set their log
probs to `inf` so that they are sampled at their corresponding index. probs to `inf` so that they are sampled at their corresponding index.
See [the paper](https://arxiv.org/abs/2212.04356) for more information.
Args: Args:
generate_config (`GenerateConfig`): generate_config (`GenerateConfig`):
The generate config used to generate the output. The following parameters are required: The generate config used to generate the output. The following parameters are required:
...@@ -1399,6 +1401,8 @@ class ClassifierFreeGuidanceLogitsProcessor(LogitsProcessor): ...@@ -1399,6 +1401,8 @@ class ClassifierFreeGuidanceLogitsProcessor(LogitsProcessor):
correspond to the unconditional logits (predicted from an empty or 'null' prompt). The processor computes a correspond to the unconditional logits (predicted from an empty or 'null' prompt). The processor computes a
weighted average across the conditional and unconditional logits, parameterised by the `guidance_scale`. weighted average across the conditional and unconditional logits, parameterised by the `guidance_scale`.
See [the paper](https://arxiv.org/abs/2306.05284) for more information.
Args: Args:
guidance_scale (float): guidance_scale (float):
The guidance scale for classifier free guidance (CFG). CFG is enabled by setting `guidance_scale > 1`. The guidance scale for classifier free guidance (CFG). CFG is enabled by setting `guidance_scale > 1`.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment