Aneeth commited on
Commit
45bb491
·
verified ·
1 Parent(s): 929735b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  from peft import AutoPeftModelForCausalLM
2
  from transformers import GenerationConfig
3
- from transformers import AutoTokenizer
4
  import torch
5
  import streamlit as st
6
 
 
1
  from peft import AutoPeftModelForCausalLM
2
  from transformers import GenerationConfig
3
+ from transformers import AutoTokenizer, GPTQConfig
4
  import torch
5
  import streamlit as st
6