kanneboinakumar commited on
Commit
6a6e5b8
·
verified ·
1 Parent(s): 68a69a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
5
  from peft import PeftModel
6
 
7
  # Load Base Gemma Model (Required for Adapter)
8
- base_model_path = "google/gemma-1b"
9
  model = AutoModelForCausalLM.from_pretrained(
10
  base_model_path,
11
  load_in_4bit=True, # Efficient memory usage
 
5
  from peft import PeftModel
6
 
7
  # Load Base Gemma Model (Required for Adapter)
8
+ base_model_path = "google/gemma-3-1b-it"
9
  model = AutoModelForCausalLM.from_pretrained(
10
  base_model_path,
11
  load_in_4bit=True, # Efficient memory usage