fix: sampling code example typo

Author: flavioschneider

README.md

@@ -61,7 +61,7 @@ unet = UNet1d(
 x = torch.randn(3, 1, 2 ** 16)
 t = torch.tensor([0.2, 0.8, 0.3])
 
-y = unet(x, t) # [2, 1, 32768], 2 samples of ~1.5 seconds of generated audio at 22050kHz
+y = unet(x, t) # [3, 1, 32768], compute 3 samples of ~1.5 seconds at 22050Hz with the given noise levels t
 ```
 
 ### Diffusion
@@ -84,7 +84,7 @@ loss.backward() # Do this many times
 
 #### Sampling
 ```python
-from audio_diffusion_pytorch import DiffusionSampler, KerrasSchedule
+from audio_diffusion_pytorch import DiffusionSampler, KarrasSchedule
 
 sampler = DiffusionSampler(
     diffusion,