nits cleanup

modal-labs · Jul 26, 2023 · 9afe394 · 9afe394
1 parent 9cccea8
commit 9afe394
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/06_gpu_and_ml/openai_whisper/pod_transcriber/app/main.py b/06_gpu_and_ml/openai_whisper/pod_transcriber/app/main.py
@@ -52,7 +52,7 @@
     secrets=[Secret.from_name("podchaser")],
 )
 
-stub.in_progress = Dict()
+stub.in_progress = Dict.new()
 
 
 def utc_now() -> datetime.datetime:

diff --git a/misc/gpt2_language_model.py b/misc/gpt2_language_model.py
@@ -5,10 +5,10 @@
 
 stub = modal.Stub("example-gpt2")
 
-# volume = modal.NetworkFileSystem().persist("gpt2")
 CACHE_PATH = "/root/model_cache"
 
 
+# Run as a build function to save the model files into the custom `modal.Image`.
 def download_model():
     from transformers import pipeline