sanchit-gandhi HF staff commited on
Commit
c3b2245
1 Parent(s): c691cea

Add model weights and README

Browse files
Files changed (2) hide show
  1. .idea/workspace.xml +3 -2
  2. README.md +1 -0
.idea/workspace.xml CHANGED
@@ -5,7 +5,8 @@
5
  </component>
6
  <component name="ChangeListManager">
7
  <list default="true" id="a184a98c-762c-486e-837a-90fb6e4a8da7" name="Changes" comment="">
8
- <change afterPath="$PROJECT_DIR$/README.md" afterDir="false" />
 
9
  </list>
10
  <option name="SHOW_DIALOG" value="false" />
11
  <option name="HIGHLIGHT_CONFLICTS" value="true" />
@@ -46,7 +47,7 @@
46
  <option name="presentableId" value="Default" />
47
  <updated>1698836178917</updated>
48
  <workItem from="1698836180167" duration="14374000" />
49
- <workItem from="1698919507129" duration="5086000" />
50
  </task>
51
  <servers />
52
  </component>
 
5
  </component>
6
  <component name="ChangeListManager">
7
  <list default="true" id="a184a98c-762c-486e-837a-90fb6e4a8da7" name="Changes" comment="">
8
+ <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
9
+ <change beforePath="$PROJECT_DIR$/README.md" beforeDir="false" afterPath="$PROJECT_DIR$/README.md" afterDir="false" />
10
  </list>
11
  <option name="SHOW_DIALOG" value="false" />
12
  <option name="HIGHLIGHT_CONFLICTS" value="true" />
 
47
  <option name="presentableId" value="Default" />
48
  <updated>1698836178917</updated>
49
  <workItem from="1698836180167" duration="14374000" />
50
+ <workItem from="1698919507129" duration="6356000" />
51
  </task>
52
  <servers />
53
  </component>
README.md CHANGED
@@ -292,6 +292,7 @@ model_id = "distil-whisper/distil-large-v2"
292
 
293
  # load the model + processor
294
  model = AutoModelForSpeechSeq2Seq.from_pretrained(model_id, torch_dtype=torch_dtype, use_safetensors=True, low_cpu_mem_usage=True)
 
295
  processor = AutoProcessor.from_pretrained(model_id)
296
 
297
  # load the dataset with streaming mode
 
292
 
293
  # load the model + processor
294
  model = AutoModelForSpeechSeq2Seq.from_pretrained(model_id, torch_dtype=torch_dtype, use_safetensors=True, low_cpu_mem_usage=True)
295
+ model = model.to(device)
296
  processor = AutoProcessor.from_pretrained(model_id)
297
 
298
  # load the dataset with streaming mode