add assert
This commit is contained in:
parent
1a0c63edaf
commit
35caccd376
|
|
@ -57,6 +57,9 @@ if ft_config.local_rank == 0:
|
|||
if ft_config.gradient_checkpointing:
|
||||
print('Disable Dropout.')
|
||||
|
||||
if ft_config.mbatch_size > ft_config.batch_size:
|
||||
raise Exception('batch_size need to be larger than mbatch_size.')
|
||||
|
||||
# Load Basic Model
|
||||
model, tokenizer = load_llama_model_4bit_low_ram(ft_config.llama_q4_config_dir,
|
||||
ft_config.llama_q4_model,
|
||||
|
|
|
|||
Loading…
Reference in New Issue