"docs/index.rst" did not exist on "3282bbb53d5676003b3e392ec893465144feaf4c"
  • Matthew Douglas's avatar
    4bit quantization for arbitrary `nn.Parameter` (#1720) · 27549fb0
    Matthew Douglas authored
    * Add parametrize util for targeting parameters outside of nn.Linear modules
    
    * Parametrize 4bit: replace existing prequantized weight
    
    * cleanup
    
    * Add caching for parametrization
    
    * Add tests
    
    * Fix tests
    
    * Guard for torch < 2.5
    
    * Guard for torch < 2.5
    
    * Another test gaurd for torch >= 2.5
    27549fb0
test_parametrize.py 19.4 KB