take avg_diff out of attributes and save to cpu (#1) 6ef419c verified linoyts HF staff commited on Jul 25
revert taking out avg_diff from attributes part 2 b98069b verified linoyts HF staff commited on Jul 25