• Matt's avatar
    Fix tflongformer int dtype (#18907) · c126a239
    Matt authored
    * Use int64 throughout TFLongFormer
    
    * make style
    
    * Do some more fixed casting in TFLongFormer
    
    * Fix some wonky "is None" conditionals
    
    * Cast all the dtypes, salt the earth
    
    * Fix copies to TFLED as well and do some casting there
    
    * dtype fix in TFLongformer test
    
    * Make fixup
    
    * Expand tolerances on the LED tests too (I think this is a TF32 thing)
    
    * Expand test tolerances for LED a tiny bit (probably a Tensorfloat thing again)
    c126a239
test_modeling_tf_led.py 18.3 KB