Update app.py
Browse files
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
|