Spaces:
Running
Running
Update CLIP.py
Browse files
CLIP.py
CHANGED
@@ -1,6 +1,8 @@
|
|
1 |
from transformers import CLIPTextModelWithProjection, CLIPTokenizer
|
2 |
import torch
|
3 |
from safetensors.torch import load_file as load_safetensor
|
|
|
|
|
4 |
|
5 |
# Device configuration
|
6 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
|
|
1 |
from transformers import CLIPTextModelWithProjection, CLIPTokenizer
|
2 |
import torch
|
3 |
from safetensors.torch import load_file as load_safetensor
|
4 |
+
from diffusers import AutoencoderKL
|
5 |
+
|
6 |
|
7 |
# Device configuration
|
8 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|