search

HazyResearch / H3

Language Modeling with the H3 State Space Model
Apache License 2.0
511 stars 53 forks source link

ssm utils #2

Open bryanhpchiang opened 1 year ago

bryanhpchiang commented 1 year ago

trying to run the benchmark -- where are the ssm_utils mentioned here?https://github.com/HazyResearch/H3/blob/main/src/models/ssm/ss_kernel_diag.py#L15

thanks!

DanFu09 commented 1 year ago

Pushed now! Let us know if you run into any other problems.

bryanhpchiang commented 1 year ago

works! running into the error in #3 but i think that's an OOM issue