Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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"
|