zhiweili
commited on
Commit
·
8772d2d
1
Parent(s):
452f3b5
import load_image
Browse files- app_tensorrt.py +2 -1
app_tensorrt.py
CHANGED
|
@@ -7,9 +7,10 @@ from diffusers import (
|
|
| 7 |
DDPMScheduler,
|
| 8 |
StableDiffusionXLImg2ImgPipeline,
|
| 9 |
AutoencoderKL,
|
| 10 |
-
load_image,
|
| 11 |
)
|
| 12 |
|
|
|
|
|
|
|
| 13 |
os.system("pip install torch_tensorrt==2.4.0")
|
| 14 |
|
| 15 |
BASE_MODEL = "stabilityai/stable-diffusion-xl-base-1.0"
|
|
|
|
| 7 |
DDPMScheduler,
|
| 8 |
StableDiffusionXLImg2ImgPipeline,
|
| 9 |
AutoencoderKL,
|
|
|
|
| 10 |
)
|
| 11 |
|
| 12 |
+
from diffusers.utils import load_image
|
| 13 |
+
|
| 14 |
os.system("pip install torch_tensorrt==2.4.0")
|
| 15 |
|
| 16 |
BASE_MODEL = "stabilityai/stable-diffusion-xl-base-1.0"
|