Update src/pipeline.py
Browse files- src/pipeline.py +1 -0
src/pipeline.py
CHANGED
|
@@ -41,6 +41,7 @@ from diffusers.models.embeddings import CombinedTimestepGuidanceTextProjEmbeddin
|
|
| 41 |
from diffusers.models.modeling_outputs import Transformer2DModelOutput
|
| 42 |
|
| 43 |
from model import E, D
|
|
|
|
| 44 |
|
| 45 |
import os
|
| 46 |
os.environ['PYTORCH_CUDA_ALLOC_CONF']="expandable_segments:True"
|
|
|
|
| 41 |
from diffusers.models.modeling_outputs import Transformer2DModelOutput
|
| 42 |
|
| 43 |
from model import E, D
|
| 44 |
+
import torchvision
|
| 45 |
|
| 46 |
import os
|
| 47 |
os.environ['PYTORCH_CUDA_ALLOC_CONF']="expandable_segments:True"
|