There are significant amount of codes specifically written for the model tiiuae/falcon-mamba-tiny-dev
if config.__class__.__name__ == "FalconMambaConfig":
pass
|
if config.__class__.__name__ == "FalconMambaConfig": |
We should simplify/merge it to the main stream of code if possible.
cc @xadupre
There are significant amount of codes specifically written for the model
tiiuae/falcon-mamba-tiny-devonnx-diagnostic/onnx_diagnostic/tasks/text_generation.py
Line 30 in f86c55d
We should simplify/merge it to the main stream of code if possible.
cc @xadupre