Edit 536 lines due to type mismatch error when evaluation
#14
by
						
shoveling42
	
							
						- opened
							
					
The specific error message is as follows:
RuntimeError: Index put requires the source and destination dtypes match, got BFloat16 for the destination and Float for the source.
