Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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)
|