captainkyd commited on
Commit
e0e2048
1 Parent(s): d0a36d2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -2,7 +2,8 @@ import spaces
2
  import gradio as gr
3
  import torch
4
  import transformers
5
- from transformers import AutoModelForCausalLM, AutoTokenizer
 
6
  import os
7
 
8
  title = """# Welcome to 🌟Tonic's🐇🥷🏻Trinity
 
2
  import gradio as gr
3
  import torch
4
  import transformers
5
+ from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
6
+ import accelerate
7
  import os
8
 
9
  title = """# Welcome to 🌟Tonic's🐇🥷🏻Trinity