Commit
•
c1f8208
1
Parent(s):
7bfd86c
testing with new model
Browse files
app.py
CHANGED
|
@@ -129,7 +129,9 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
| 129 |
import torch
|
| 130 |
import gradio as gr
|
| 131 |
|
| 132 |
-
model_id = "HuggingFaceTB/SmolLM2-135M"
|
|
|
|
|
|
|
| 133 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
| 134 |
model = AutoModelForCausalLM.from_pretrained(model_id)
|
| 135 |
|
|
|
|
| 129 |
import torch
|
| 130 |
import gradio as gr
|
| 131 |
|
| 132 |
+
# model_id = "HuggingFaceTB/SmolLM2-135M"
|
| 133 |
+
model_id = "MaxBlumenfeld/smollm2-135m-bootleg-instruct01"
|
| 134 |
+
|
| 135 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
| 136 |
model = AutoModelForCausalLM.from_pretrained(model_id)
|
| 137 |
|