Spaces:
Running
on
Zero
Running
on
Zero
Update models/unet.py
Browse files- models/unet.py +0 -1
models/unet.py
CHANGED
|
@@ -865,7 +865,6 @@ class MotionCLR(nn.Module):
|
|
| 865 |
|
| 866 |
def encode_text(self, raw_text, device):
|
| 867 |
self.clip_model.token_embedding = self.clip_model.token_embedding.to(device)
|
| 868 |
-
self.clip_model.positional_embedding = self.clip_model.positional_embedding.to(device)
|
| 869 |
self.clip_model.transformer = self.clip_model.transformer.to(device)
|
| 870 |
self.clip_model.ln_final = self.clip_model.ln_final.to(device)
|
| 871 |
with torch.no_grad():
|
|
|
|
| 865 |
|
| 866 |
def encode_text(self, raw_text, device):
|
| 867 |
self.clip_model.token_embedding = self.clip_model.token_embedding.to(device)
|
|
|
|
| 868 |
self.clip_model.transformer = self.clip_model.transformer.to(device)
|
| 869 |
self.clip_model.ln_final = self.clip_model.ln_final.to(device)
|
| 870 |
with torch.no_grad():
|