kazuma313 commited on
Commit
d74609a
1 Parent(s): 8ced6b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=model, offload_dir=save_dir)
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