Cant run hybrid model

[usamacloudops]

AssertionError Traceback (most recent call last)
in <cell line: 0>()
7
8 model = Zonos.from_pretrained("Zyphra/Zonos-v0.1-transformer", device=device)
----> 9 model2 = Zonos.from_pretrained("Zyphra/Zonos-v0.1-hybrid", device=device)

3 frames
/content/Zonos/zonos/backbone/_torch.py in init(self, config)
55
56 def init(self, config: BackboneConfig):
---> 57 assert not config.ssm_cfg, "This backbone implementation only supports the Transformer model."
58 super().init()
59 self.config = config

AssertionError: This backbone implementation only supports the Transformer model.

Trying to run the Zyphra/Zonos-v0.1-hybrid but getting above error, any idea how to solve it

[sekstini]

Hey, if you have a recent Nvidia GPU (3000-series or newer) you can install the optional dependencies as shown here. This will enable you to run the Hybrid model.

If not then you will only be able to run the Transformer variant.