sarashina2.1-1b-sft-CPU

Runtime error

aixsatoshi commited on Dec 11, 2024

Commit

d139d80

verified ·

1 Parent(s): a352461

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,17 +6,17 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStream
 import gradio as gr
 from threading import Thread
-model_id = "ibm-granite/granite-3.0-1b-a400m-instruct"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(
   model_id,
   device_map="cpu",
 )
-TITLE = "<h1><center>ibm-granite/granite-3.0-1b-a400m-instruct by CPU</center></h1>"
 DESCRIPTION = """
-<h3>MODEL: <a href="https://huggingface.co/ibm-granite/granite-3.0-1b-a400m-instruct">ibm-granite/granite-3.0-1b-a400m-instruct</a></h3>
 <center>
 <p>This model is designed for conversational interactions.</p>
 </center>

 import gradio as gr
 from threading import Thread
+model_id = "Aratako/sarashina2.1-1b-sft"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(
   model_id,
   device_map="cpu",
 )
+TITLE = "<h1><center>Aratako/sarashina2.1-1b-sft by CPU</center></h1>"
 DESCRIPTION = """
+<h3>MODEL: <a href="https://huggingface.co/Aratako/sarashina2.1-1b-sft">Aratako/sarashina2.1-1b-sft</a></h3>
 <center>
 <p>This model is designed for conversational interactions.</p>
 </center>