Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -14,7 +14,7 @@ login(token=hf_token, add_to_git_credential=True)
|
|
14 |
|
15 |
from accelerate import disk_offload
|
16 |
save_dir="gemma-dokter-ft"
|
17 |
-
disk_offload(model=
|
18 |
|
19 |
# config = PeftConfig.from_pretrained(peft_model)
|
20 |
|
|
|
14 |
|
15 |
from accelerate import disk_offload
|
16 |
save_dir="gemma-dokter-ft"
|
17 |
+
disk_offload(model=peft_model, offload_dir=save_dir)
|
18 |
|
19 |
# config = PeftConfig.from_pretrained(peft_model)
|
20 |
|