Ankitajadhav commited on
Commit
cd85695
·
verified ·
1 Parent(s): a72e07a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -3
app.py CHANGED
@@ -13,9 +13,6 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, Trainer, TrainingA
13
  # Set environment variables to address warnings
14
  os.environ['TF_ENABLE_ONEDNN_OPTS'] = '0'
15
 
16
- # Ensure necessary packages are installed
17
- !pip install accelerate
18
- !pip install flash-attention
19
 
20
  torch.random.manual_seed(0)
21
  model_name = "microsoft/Phi-3-mini-4k-instruct"
 
13
  # Set environment variables to address warnings
14
  os.environ['TF_ENABLE_ONEDNN_OPTS'] = '0'
15
 
 
 
 
16
 
17
  torch.random.manual_seed(0)
18
  model_name = "microsoft/Phi-3-mini-4k-instruct"