sc2582 commited on
Commit
591016c
·
verified ·
1 Parent(s): f6c9f8c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ 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-8B-Instruct-sft-alfworld-iter0"
7
  #peft_model_id = "decision-oaif/Meta-Llama-3-8B-Instruct-sft-intercode-python-iter0"
8
  # Load model with device map and dtype
9
  model = AutoModelForCausalLM.from_pretrained(
 
3
  import gradio as gr
4
 
5
  # Adjust this to your model ID
6
+ model_id = "decision-oaif/Meta-Llama-3-8B-Instruct-sft-alfworld-iter1"
7
  #peft_model_id = "decision-oaif/Meta-Llama-3-8B-Instruct-sft-intercode-python-iter0"
8
  # Load model with device map and dtype
9
  model = AutoModelForCausalLM.from_pretrained(