Fix: interface change for sampling weights (#156) #750

Job	Run time
sample-config	13s
lint (accelerated-moe)	4m 51s
lint (online-data-mixing)	5m 22s
lint (accelerated-peft)	5m 10s
lint (fused-ops-and-kernels)	4m 44s
lint (attention-and-distributed-packing)	5m 48s
lint (framework)	4m 55s
	31m 3s

Navigation Menu