sc2582 commited on
Commit
e6d9ae4
·
verified ·
1 Parent(s): fe95092

Update app.py

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