MaxBlumenfeld
commited on
Commit
·
96bd27b
1
Parent(s):
c1f8208
changed to use model with config file
Browse files
app.py
CHANGED
@@ -130,7 +130,7 @@ import torch
|
|
130 |
import gradio as gr
|
131 |
|
132 |
# model_id = "HuggingFaceTB/SmolLM2-135M"
|
133 |
-
model_id = "MaxBlumenfeld/smollm2-135m-bootleg-
|
134 |
|
135 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
136 |
model = AutoModelForCausalLM.from_pretrained(model_id)
|
|
|
130 |
import gradio as gr
|
131 |
|
132 |
# model_id = "HuggingFaceTB/SmolLM2-135M"
|
133 |
+
model_id = "MaxBlumenfeld/smollm2-135m-bootleg-instruct02"
|
134 |
|
135 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
136 |
model = AutoModelForCausalLM.from_pretrained(model_id)
|