sc2582 commited on
Commit
42b72d5
·
verified ·
1 Parent(s): fba1e5d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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-iter0"
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,