Walmart-the-bag commited on
Commit
91d1248
Β·
verified Β·
1 Parent(s): 91b99e6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from transformers import TextIteratorStreamer
3
  from threading import Thread
4
  from transformers import StoppingCriteria, StoppingCriteriaList
5
  import torch
6
-
7
  model_name = "microsoft/Phi-3-mini-128k-instruct"
8
  from transformers import AutoModelForCausalLM, AutoTokenizer
9
  tokenizer = AutoTokenizer.from_pretrained(model_name)
 
3
  from threading import Thread
4
  from transformers import StoppingCriteria, StoppingCriteriaList
5
  import torch
6
+ import spaces
7
  model_name = "microsoft/Phi-3-mini-128k-instruct"
8
  from transformers import AutoModelForCausalLM, AutoTokenizer
9
  tokenizer = AutoTokenizer.from_pretrained(model_name)