jalonso24 commited on
Commit
d657256
·
verified ·
1 Parent(s): 2029bcf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -10,8 +10,10 @@ adapter_path = "jalonso24/lora-lateblight-v3"
10
  base_model = AutoModelForCausalLM.from_pretrained(
11
  base_model_id,
12
  torch_dtype=torch.float32,
13
- device_map="auto"
 
14
  )
 
15
  model = PeftModel.from_pretrained(base_model, adapter_path)
16
 
17
  tokenizer = AutoTokenizer.from_pretrained(base_model_id, use_fast=False)
 
10
  base_model = AutoModelForCausalLM.from_pretrained(
11
  base_model_id,
12
  torch_dtype=torch.float32,
13
+ device_map="auto",
14
+ offload_folder="./offload" # 👈 tells it where to put overflow weights
15
  )
16
+
17
  model = PeftModel.from_pretrained(base_model, adapter_path)
18
 
19
  tokenizer = AutoTokenizer.from_pretrained(base_model_id, use_fast=False)