adding os
Browse files- S2I/modules/models.py +1 -0
S2I/modules/models.py
CHANGED
@@ -1,5 +1,6 @@
|
|
1 |
import torch
|
2 |
import copy
|
|
|
3 |
from diffusers import DDPMScheduler
|
4 |
from transformers import AutoTokenizer, CLIPTextModel
|
5 |
from diffusers import AutoencoderKL, UNet2DConditionModel
|
|
|
1 |
import torch
|
2 |
import copy
|
3 |
+
import os
|
4 |
from diffusers import DDPMScheduler
|
5 |
from transformers import AutoTokenizer, CLIPTextModel
|
6 |
from diffusers import AutoencoderKL, UNet2DConditionModel
|