ProCreations commited on
Commit
e4b0b00
·
verified ·
1 Parent(s): 23814a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ import torch; torch.set_num_threads(os.cpu_count()); torch.set_num_interop_threa
10
  import gradio as gr
11
  from transformers import AutoTokenizer, AutoModelForCausalLM
12
 
13
- MODEL_NAME = "meta-llama/Llama-3.1-8B-Instruct"
14
  PROMPTS_PATH = "full_prompts.json"
15
  STATE_PATH = "current_state.json"
16
  DATA_PATH = "data.json"
 
10
  import gradio as gr
11
  from transformers import AutoTokenizer, AutoModelForCausalLM
12
 
13
+ MODEL_NAME = "meta-llama/Llama-3.2-3B-Instruct"
14
  PROMPTS_PATH = "full_prompts.json"
15
  STATE_PATH = "current_state.json"
16
  DATA_PATH = "data.json"