Skip to content

Commit

Permalink
Deploying to gh-pages from @ c04d82f 🚀
Browse files Browse the repository at this point in the history
  • Loading branch information
EricLBuehler committed Apr 2, 2024
1 parent 01d2015 commit b58b62b
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 6 deletions.
8 changes: 4 additions & 4 deletions src/candle_sampling/logits_processor.rs.html
Original file line number Diff line number Diff line change
Expand Up @@ -657,9 +657,9 @@ <h2>Files</h2></div></nav><div class="sidebar-resizer"></div>
<span class="kw">fn </span>sample_topk(<span class="kw-2">&amp;mut </span><span class="self">self</span>, probs: <span class="kw-2">&amp;mut </span>Vec&lt;f32&gt;, top_k: usize) -&gt; <span class="prelude-ty">Result</span>&lt;Logprobs&gt; {
<span class="comment">// Sort probs into descending order (highest probs first)
</span>probs.sort_by(|x, y| x.total_cmp(y));
probs.reverse();
<span class="comment">//probs.reverse();

<span class="comment">// Clamp smaller probabilities to zero.
// Clamp smaller probabilities to zero.
</span><span class="kw">for </span>(index, val) <span class="kw">in </span>probs.iter_mut().enumerate() {
<span class="kw">if </span>index &gt;= top_k {
<span class="kw-2">*</span>val = <span class="number">0.0</span>;
Expand All @@ -673,9 +673,9 @@ <h2>Files</h2></div></nav><div class="sidebar-resizer"></div>
<span class="kw">fn </span>sample_topkp(<span class="kw-2">&amp;mut </span><span class="self">self</span>, probs: <span class="kw-2">&amp;mut </span>Vec&lt;f32&gt;, top_k: usize, top_p: f32) -&gt; <span class="prelude-ty">Result</span>&lt;Logprobs&gt; {
<span class="comment">// Sort probs into descending order (highest probs first)
</span>probs.sort_by(|x, y| x.total_cmp(y));
probs.reverse();
<span class="comment">//probs.reverse();

<span class="comment">// TOP K
// TOP K
// Clamp smaller probabilities to zero.
</span><span class="kw">for </span>(index, val) <span class="kw">in </span>probs.iter_mut().enumerate() {
<span class="kw">if </span>index &gt;= top_k {
Expand Down
2 changes: 1 addition & 1 deletion trait.impl/core/clone/trait.Clone.js

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

2 changes: 1 addition & 1 deletion trait.impl/serde/ser/trait.Serialize.js

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

0 comments on commit b58b62b

Please sign in to comment.