sc2582 commited on
Commit
f6c9f8c
·
verified ·
1 Parent(s): 8a04b2a

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 = "meta-llama/Meta-Llama-3.1-8B-Instruct"
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-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(