starnernj commited on
Commit
35fac3e
·
verified ·
1 Parent(s): a84397e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -20,9 +20,10 @@ login(token=os.getenv("HuggingFaceFineGrainedReadToken"))
20
 
21
  model = AutoModelForCausalLM.from_pretrained(
22
  base_model_name,
23
- load_in_4bit=True, # ✅ Reduces memory
24
  torch_dtype=torch.float16,
25
- device_map="auto"
 
26
  )
27
 
28
  # Load LoRA adapter
 
20
 
21
  model = AutoModelForCausalLM.from_pretrained(
22
  base_model_name,
23
+ # load_in_4bit=True, # ✅ Reduces memory, but requires a GPU
24
  torch_dtype=torch.float16,
25
+ # device_map="auto" # Since I'm running on a free instance that doesn't have a GPU, I'll need to force CPU
26
+ device_map="cpu"
27
  )
28
 
29
  # Load LoRA adapter