Update models/unet.py
Browse files- models/unet.py +4 -0
models/unet.py
CHANGED
|
@@ -838,6 +838,10 @@ class MotionCLR(nn.Module):
|
|
| 838 |
)
|
| 839 |
|
| 840 |
def encode_text(self, raw_text, device):
|
|
|
|
|
|
|
|
|
|
|
|
|
| 841 |
with torch.no_grad():
|
| 842 |
texts = clip.tokenize(raw_text, truncate=True).to(
|
| 843 |
device
|
|
|
|
| 838 |
)
|
| 839 |
|
| 840 |
def encode_text(self, raw_text, device):
|
| 841 |
+
print("00000000")
|
| 842 |
+
print(device)
|
| 843 |
+
print(self.clip_model.device)
|
| 844 |
+
print("00000000")
|
| 845 |
with torch.no_grad():
|
| 846 |
texts = clip.tokenize(raw_text, truncate=True).to(
|
| 847 |
device
|