JPQ24 commited on
Commit
236ed28
·
verified ·
1 Parent(s): 5c5ab7e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,8 +10,8 @@ import os
10
  # ---------------------------------------------------------------------------
11
  # WARNING: On CPU, 8B models are very heavy.
12
  # If this crashes, switch to "unsloth/Llama-3.2-3B-Instruct"
13
- BASE_MODEL_ID = "unsloth/llama-3-8b-Instruct-bnb-4bit"
14
- LORA_ADAPTER_ID = "JPQ24/llama-3-8b-Natural-synthesis-Lora"
15
 
16
  # ---------------------------------------------------------------------------
17
  # LOAD MODEL (State Initialization)
 
10
  # ---------------------------------------------------------------------------
11
  # WARNING: On CPU, 8B models are very heavy.
12
  # If this crashes, switch to "unsloth/Llama-3.2-3B-Instruct"
13
+ BASE_MODEL_ID = "unsloth/Llama-3.2-1B-Instruct-bnb-4bit"
14
+ LORA_ADAPTER_ID = "JPQ24/Natural-synthesis-llama-3.2-1b"
15
 
16
  # ---------------------------------------------------------------------------
17
  # LOAD MODEL (State Initialization)