Add emb_patch support to UNetModel forward (#4779) #123

Job	Run time
test-stable (macos, 3.9, 12.1, stable)	9m 52s
test-stable (macos, 3.10, 12.1, stable)	9m 35s
test-unix-nightly (macos, 3.11, 12.1, nightly)	10m 27s
test-stable (macos, 3.11, 12.1, stable)	9m 58s
test-unix-nightly (linux, 3.11, 12.1, nightly)	5m 53s
test-stable (macos, 3.12, 12.1, stable)	9m 54s
test-stable (linux, 3.9, 12.1, stable)	5m 59s
test-stable (linux, 3.10, 12.1, stable)	5m 58s
test-stable (linux, 3.11, 12.1, stable)	5m 58s
test-win-nightly (windows, 3.9, 12.1, nightly)	3m 41s
test-win-nightly (windows, 3.10, 12.1, nightly)	3m 52s
test-stable (linux, 3.12, 12.1, stable)	5m 57s
test-win-nightly (windows, 3.11, 12.1, nightly)	3m 43s
test-stable (windows, 3.9, 12.1, stable)	3m 39s
test-win-nightly (windows, 3.12, 12.1, nightly)	3m 46s
test-stable (windows, 3.10, 12.1, stable)	3m 39s
test-stable (windows, 3.11, 12.1, stable)	3m 38s
test-stable (windows, 3.12, 12.1, stable)	3m 40s
	1h 49m 9s

Provide feedback