• LuGY's avatar
    Add CI and inference unit test (#85) · 441bae7f
    LuGY authored
    * add CI, update Dockerfile
    
    * remove useless loop in inference, add some comments to Attention
    
    * update inference test and CI
    
    * fix path
    
    * add pytest for test env
    
    * add einops install
    
    * add cuda cache, loose the condition
    441bae7f
inference.py 19 KB