Update README.md
#2
by
						
muhtasham
	
							
						- opened
							
					
    	
        README.md
    CHANGED
    
    | 
         @@ -42,7 +42,7 @@ device = "cuda:0" if torch.cuda.is_available() else "cpu" 
     | 
|
| 42 | 
         
             
            compute_type = "float16" if torch.cuda.is_available() else "float32"
         
     | 
| 43 | 
         | 
| 44 | 
         
             
            # load model on GPU if available, else cpu
         
     | 
| 45 | 
         
            -
            model = WhisperModel("distil-large-v3", device=device, compute_type=compute_type)
         
     | 
| 46 | 
         | 
| 47 | 
         
             
            # load toy dataset for example
         
     | 
| 48 | 
         
             
            dataset = load_dataset("hf-internal-testing/librispeech_asr_dummy", "clean", split="validation")
         
     | 
| 
         | 
|
| 42 | 
         
             
            compute_type = "float16" if torch.cuda.is_available() else "float32"
         
     | 
| 43 | 
         | 
| 44 | 
         
             
            # load model on GPU if available, else cpu
         
     | 
| 45 | 
         
            +
            model = WhisperModel("distil-whisper/distil-large-v3-ct2", device=device, compute_type=compute_type)
         
     | 
| 46 | 
         | 
| 47 | 
         
             
            # load toy dataset for example
         
     | 
| 48 | 
         
             
            dataset = load_dataset("hf-internal-testing/librispeech_asr_dummy", "clean", split="validation")
         
     |