huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 11:08:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Display the temperature being used for text generation. (#278)

Browse Source
This commit is contained in:
Laurent Mazare
2023-07-30 09:53:05 +01:00
committed by GitHub
parent c950a5c6b1
commit ba2254556c
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
candle-wasm-examples/llama2-c/src/app.rs
View File

@ -30,6 +30,7 @@ async fn fetch_url(url: &str) -> Result<Vec<u8>, JsValue> {
} }
pub enum Msg { pub enum Msg {
Refresh,
Run, Run,
UpdateStatus(String), UpdateStatus(String),
SetModel(ModelData), SetModel(ModelData),
@ -166,18 +167,20 @@ impl Component for App {
self.status = status; self.status = status;
true true
} }
Msg::Refresh => true,
} }
} }
fn view(&self, ctx: &Context<Self>) -> Html { fn view(&self, ctx: &Context<Self>) -> Html {
use yew::TargetCast; use yew::TargetCast;
let temperature = self.temperature.clone(); let temperature = self.temperature.clone();
let oninput = move |e: yew::InputEvent| { let oninput = ctx.link().callback(move |e: yew::InputEvent| {
let input: web_sys::HtmlInputElement = e.target_unchecked_into(); let input: web_sys::HtmlInputElement = e.target_unchecked_into();
if let Ok(temp) = f64::from_str(&input.value()) { if let Ok(temp) = f64::from_str(&input.value()) {
*temperature.borrow_mut() = temp *temperature.borrow_mut() = temp
} }
}; Msg::Refresh
});
html! { html! {
<div style="margin: 2%;"> <div style="margin: 2%;">
<div><p>{"Running "} <div><p>{"Running "}
@ -188,8 +191,9 @@ impl Component for App {
<p>{"Once the weights have loaded, click on the run button to start generating content."} <p>{"Once the weights have loaded, click on the run button to start generating content."}
</p> </p>
</div> </div>
{"temperature: "}<input type="range" min="0." max="1.2" step="0.1" value={self.temperature.borrow().to_string()} {oninput} id="temp"/> {"temperature \u{00a0} "}
<input type="range" min="0." max="1.2" step="0.1" value={self.temperature.borrow().to_string()} {oninput} id="temp"/>
{format!(" \u{00a0} {}", self.temperature.borrow())}
<br/ > <br/ >
{ {
if self.loaded{ if self.loaded{