"benchmark/vscode:/vscode.git/clone" did not exist on "61e7c4dd21f5414d9c50f8ab6c670b3d3332875e"
  • Suraj Patil's avatar
    update flax scheduler API (#822) · 0a09af2f
    Suraj Patil authored
    * update flax scheduler API
    
    * remoev set format
    
    * fix call to scale_model_input
    
    * update flax pndm
    
    * use int32
    
    * update docstr
    0a09af2f
scheduling_ddim_flax.py 12.7 KB