• Michael Benayoun's avatar
    Traced models serialization and torchscripting fix (#17206) · 2e7e4280
    Michael Benayoun authored
    * Fix torch.jit.script and pickling issues
    
    * Fix get_attr issues
    
    * Fix import in function
    
    * Fix GPT-J and T5 tracing for torch=1.11
    
    * Gate graph surgery on torch version
    
    * Modeling minor changes to enable TorchScripting
    
    * Model serialization / deserialization test
    
    * Remove _assert_is_none users
    2e7e4280
test_modeling_common.py 120 KB