File size: 235 Bytes
844e150
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
tuning_config = dict(
    lora=True,
    lora_target_modules=["q_proj", "k_proj", "v_proj", "o_proj", "to_q", "to_kv", "to_out", "ff.1", "ff.3"],
    lora_r=16,
    lora_alpha=16,
    lora_dropout=0.0,
    vis=True,
    unfrozen=[],
)