Get backend - 1

src/pipeline.py

@@ -11,6 +11,7 @@ from torch import Generator
 from torchao.quantization import quantize_, int8_weight_only
 from transformers import T5EncoderModel, CLIPTextModel, logging 
 import torch._dynamo
+import torch_tensorr
 torch._dynamo.config.suppress_errors = True
 
 Pipeline: TypeAlias = FluxPipeline