removing model.cuda and device=0 since accelerate takes care of this f7c4dd5 verified jay0911 commited on Jul 31