Hugging Face's logo

Spaces:
Ankitajadhav
/
Whats_Cooking
Runtime error

App Files Community
Ankitajadhav commited on
Commit
08b8914
·
verified ·
1 Parent(s): 14fc019

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,8 +12,8 @@ import torch
12
 
13
 
14
  # Load model directly
15
- from transformers import AutoTokenizer, AutoModelForCausalLM
16
-
17
  torch.random.manual_seed(0)
18
  model = AutoModelForCausalLM.from_pretrained(
19
  "microsoft/Phi-3-mini-4k-instruct",
 
12
 
13
 
14
  # Load model directly
15
+ from transformers import AutoTokenizer, AutoModelForCausalLM, Trainer, TrainingArguments
16
+ os.environ['TF_ENABLE_ONEDNN_OPTS'] = '0'
17
  torch.random.manual_seed(0)
18
  model = AutoModelForCausalLM.from_pretrained(
19
  "microsoft/Phi-3-mini-4k-instruct",