mirror of
https://github.com/SakanaAI/doc-to-lora.git
synced 2026-07-23 17:01:04 +02:00
fix eval
This commit is contained in:
parent
e7a0569745
commit
41c622c734
1 changed files with 1 additions and 0 deletions
|
|
@ -370,6 +370,7 @@ def evaluate(checkpoint_path, args, split, generative):
|
|||
add_repeat_prompt=False,
|
||||
add_negative_prompt=False,
|
||||
use_kl_loss=False,
|
||||
set_format="pt",
|
||||
)
|
||||
|
||||
datasets = dict()
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue