fix torch_tensorrt

app_tensorrt.py

@@ -1,5 +1,5 @@
 import torch
-import torch_tensorrt
+import os
 
 from diffusers import (
     DDPMScheduler,
@@ -7,6 +7,11 @@ from diffusers import (
     AutoencoderKL,
 )
 
+os.system("pip install torch_tensorrt==2.4.0")
+
+
+import torch_tensorrt
+
 BASE_MODEL = "stabilityai/stable-diffusion-xl-base-1.0"
 device = "cuda"
 

requirements.txt

@@ -1,6 +1,5 @@
 gradio
-torch==2.4.0
-torch_tensorrt==2.4.0
+torch
 torchvision
 diffusers
 transformers