inLine-XJY commited on
Commit
28c2003
1 Parent(s): 8fdaa96

Update ldm/modules/encoders/modules.py

Browse files
Files changed (1) hide show
  1. ldm/modules/encoders/modules.py +1 -1
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="./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']
 
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']