Improved Diffusion Github Student Package