Spaces:
Sleeping
Sleeping
Update ldm/modules/encoders/modules.py
Browse files
ldm/modules/encoders/modules.py
CHANGED
@@ -528,7 +528,7 @@ class FrozenCLAPT5Embedder(AbstractEncoder):
|
|
528 |
|
529 |
class FrozenCLAPFLANEmbedder(AbstractEncoder):
|
530 |
"""Uses the CLAP transformer encoder for text from microsoft"""
|
531 |
-
def __init__(self, weights_path,t5version="
|
532 |
super().__init__()
|
533 |
|
534 |
model_state_dict = torch.load(weights_path, map_location=torch.device('cpu'))['model']
|
|
|
528 |
|
529 |
class FrozenCLAPFLANEmbedder(AbstractEncoder):
|
530 |
"""Uses the CLAP transformer encoder for text from microsoft"""
|
531 |
+
def __init__(self, weights_path,t5version="ldm/modules/encoders/CLAP/t5-v1_1-large", freeze=True, device="cuda", max_length=77): # clip-vit-base-patch32
|
532 |
super().__init__()
|
533 |
|
534 |
model_state_dict = torch.load(weights_path, map_location=torch.device('cpu'))['model']
|