• 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
release_bdist.yml 2.46 KB