Tonic commited on
Commit
f71bd6d
1 Parent(s): cc7b42b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -3,6 +3,7 @@ from trulens_eval.tru_custom_app import instrument
3
  from trulens_eval.feedback import Coherence, Relevance
4
  from transformers import AutoTokenizer, AutoModelForCausalLM
5
  import torch
 
6
 
7
  class YourModel:
8
  def __init__(self):
 
3
  from trulens_eval.feedback import Coherence, Relevance
4
  from transformers import AutoTokenizer, AutoModelForCausalLM
5
  import torch
6
+ import accelerate
7
 
8
  class YourModel:
9
  def __init__(self):