Commit
•
c3b2245
1
Parent(s):
c691cea
Add model weights and README
Browse files- .idea/workspace.xml +3 -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
|
|
|
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="
|
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
|