Tonic commited on
Commit
ecc69e5
1 Parent(s): 3b038c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -22,6 +22,7 @@ from transformers import AutoTokenizer, MistralForCausalLM
22
  import torch
23
  import random
24
  from textwrap import wrap
 
25
  from transformers import AutoConfig, AutoTokenizer, AutoModelForSeq2SeqLM, AutoModelForCausalLM, MistralForCausalLM
26
  from peft import PeftModel, PeftConfig
27
  import torch
 
22
  import torch
23
  import random
24
  from textwrap import wrap
25
+ import transformers
26
  from transformers import AutoConfig, AutoTokenizer, AutoModelForSeq2SeqLM, AutoModelForCausalLM, MistralForCausalLM
27
  from peft import PeftModel, PeftConfig
28
  import torch