Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -3,10 +3,9 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
3 |
import gradio as gr
|
4 |
|
5 |
# Adjust this to your model ID
|
6 |
-
model_id = "decision-oaif/Meta-Llama-3.1-8B-Instruct-sft-intercode-bash-
|
7 |
|
8 |
|
9 |
-
#peft_model_id = "decision-oaif/Meta-Llama-3.1-8B-Instruct-sft-intercode-bash-iter1"
|
10 |
# Load model with device map and dtype
|
11 |
model = AutoModelForCausalLM.from_pretrained(
|
12 |
model_id,
|
|
|
3 |
import gradio as gr
|
4 |
|
5 |
# Adjust this to your model ID
|
6 |
+
model_id = "decision-oaif/Meta-Llama-3.1-8B-Instruct-sft-intercode-bash-iter1"
|
7 |
|
8 |
|
|
|
9 |
# Load model with device map and dtype
|
10 |
model = AutoModelForCausalLM.from_pretrained(
|
11 |
model_id,
|