remove simulated preemptions

modal-labs · Jan 3, 2024 · fb51711 · fb51711
1 parent 2e563e9
commit fb51711
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/06_gpu_and_ml/flan_t5/flan_t5_finetune.py b/06_gpu_and_ml/flan_t5/flan_t5_finetune.py
@@ -76,9 +76,6 @@ def track_restarts(restart_tracker: modal.Dict) -> int:
     _allow_background_volume_commits=True,
 )
 def finetune(num_train_epochs: int = 1, size_percentage: int = 10):
-    from modal.exception import simulate_preemption
-
-    simulate_preemption(300)
     from datasets import load_dataset
     from transformers import (
         AutoModelForSeq2SeqLM,