Please make sure to pass low_cpu_mem_usage=False and device_map=None if you want to randomly initialize those weights or else make sure your checkpoint file is correct.
low_cpu_mem_usage=False
device_map=None
· Sign up or log in to comment
