mirror of
https://github.com/SakanaAI/doc-to-lora.git
synced 2026-07-23 17:01:04 +02:00
self-gen w/ closed_qa format + perceiver scripts + no strip response
This commit is contained in:
parent
d26f246937
commit
c9e958f963
6 changed files with 145 additions and 12 deletions
|
|
@ -19,7 +19,7 @@ uv run accelerate launch --main_process_port $port \
|
|||
--per_device_eval_batch_size=64 \
|
||||
--target_modules=down_proj \
|
||||
--num_blocks=8 \
|
||||
--num_self_attn_per_block=6 \
|
||||
--num_self_attn_per_block=3 \
|
||||
--n_latent_queries=208 \
|
||||
--num_pre_head_layers=1 \
|
||||
--lora_r=8 \
|
||||
|
|
|
|||
|
|
@ -19,7 +19,7 @@ uv run accelerate launch --main_process_port $port \
|
|||
--per_device_eval_batch_size=32 \
|
||||
--target_modules=down_proj \
|
||||
--num_blocks=8 \
|
||||
--num_self_attn_per_block=6 \
|
||||
--num_self_attn_per_block=3 \
|
||||
--ctx_encoder_type=per_layer_activations \
|
||||
--n_latent_queries=8 \
|
||||
--num_pre_head_layers=1 \
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue