• oahzxl's avatar
    Fix multimer kernel bug (#103) · 4d302345
    oahzxl authored
    * fix multimer bug
    
    * fix bug
    
    * remove print
    
    * fix index
    
    * fix layernorm length
    
    * update test
    
    * improve template code
    
    * remove print
    
    * polish layernorm
    4d302345
test_template_embedder.py 3.45 KB