h SSKJrJrJr SSKJr SSKJr SSKJ r J r J r SSK J r JrJr SS \S S \\S \\S \S\\ S\SS\\\44SjjrS\4Sjrg))AnyCallableOptional)torch)Model)PyTorchGradScaler PyTorchShimTorchScriptShim)convert_pytorch_default_inputsconvert_pytorch_default_outputsforwardNtorchscript_modelztorch.jit.ScriptModuleconvert_inputsconvert_outputsmixed_precision grad_scalerdevicez torch.devicereturnc hUc[nUc[n[S[XS.[ UUUUS9/SSS.S9$)aWrap a TorchScript model, so that it has the same API as Thinc models. torchscript_model: The TorchScript module. A value of `None` is also possible to construct a shim to deserialize into. convert_inputs: Function that converts inputs and gradients that should be passed to the model to Torch tensors. convert_outputs: Function that converts model outputs and gradients from Torch tensors Thinc arrays. mixed_precision: Enable mixed-precision. This changes whitelisted ops to run in half precision for better performance and lower memory use. grad_scaler: The gradient scaler to use for mixed-precision training. If this argument is set to "None" and mixed precision is enabled, a gradient scaler with the default configuration is used. device: The PyTorch device to run the model on. When this argument is set to "None", the default device for the currently active Thinc ops is used. Npytorch_script)rr)modelrrr)nInO)attrsshimsdims)r rrrr )rrrrrrs Y/home/james-whalen/.local/lib/python3.13/site-packages/thinc/layers/torchscriptwrapper.pyTorchScriptWrapper_v1r sW@79 !/T ' /'    %  rc URSn[U[5(d [S5eURSnURSnUR n[U[ RR5(d [S5e[ RRU5nURnURnURn[UUUUUUS9$)zrConvert a PyTorch wrapper to a TorchScript wrapper. The embedded PyTorch `Module` is converted to `ScriptModule`. rz6Expected PyTorchShim when converting a PyTorch wrapperrrz*PyTorchShim does not wrap a PyTorch module)rrrrr)r isinstancer ValueErrorr_modelrnnModulejitscript _grad_scaler_mixed_precisionrr ) rshimrr pytorch_modelrrrrs rpytorch_to_torchscript_wrapperr.Bs ;;q>D dK ( (QRR[[!12Nkk"34OKKM mUXX__ 5 5EFF ((7##K++O [[F %''  r!)NNNFNN)typingrrrcompatrrrrr r r pytorchwrapperr rrboolr r.r!rr4s**CC=A)-*.!/3'+ 2 892X&2h'2 2 +, 2 ^ $ 2 38_2j%r!