minhdang commited on
Commit
cf5249f
1 Parent(s): e342c1a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import spaces
2
  import gradio as gr
3
  import transformers
4
- from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
5
  import torch
6
  import os
7
 
@@ -11,7 +11,7 @@ title = """
11
  Join us : 🌟TeamTonic🌟 is always making cool demos! Join our active builder's 🛠️community 👻 [![Join us on Discord](https://img.shields.io/discord/1109943800132010065?label=Discord&logo=discord&style=flat-square)](https://discord.gg/GWpVpekp) On 🤗Huggingface:[MultiTransformer](https://huggingface.co/MultiTransformer) Math 🔍 [introspector](https://huggingface.co/introspector) On 🌐Github: [Tonic-AI](https://github.com/tonic-ai) & contribute to🌟 [Torchon](https://github.com/Tonic-AI/Torchon)🤗Big thanks to Yuvi Sharma and all the folks at huggingface for the community grant 🤗
12
  """
13
 
14
- bnb_config = BitsAndBytesConfig(load_in_4bit=True)
15
 
16
  model_id = "CohereForAI/c4ai-command-r-v01"
17
  tokenizer = AutoTokenizer.from_pretrained(model_id)
 
1
  import spaces
2
  import gradio as gr
3
  import transformers
4
+ from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig,AwqConfig
5
  import torch
6
  import os
7
 
 
11
  Join us : 🌟TeamTonic🌟 is always making cool demos! Join our active builder's 🛠️community 👻 [![Join us on Discord](https://img.shields.io/discord/1109943800132010065?label=Discord&logo=discord&style=flat-square)](https://discord.gg/GWpVpekp) On 🤗Huggingface:[MultiTransformer](https://huggingface.co/MultiTransformer) Math 🔍 [introspector](https://huggingface.co/introspector) On 🌐Github: [Tonic-AI](https://github.com/tonic-ai) & contribute to🌟 [Torchon](https://github.com/Tonic-AI/Torchon)🤗Big thanks to Yuvi Sharma and all the folks at huggingface for the community grant 🤗
12
  """
13
 
14
+ bnb_config = AwqConfig(group_size=512)
15
 
16
  model_id = "CohereForAI/c4ai-command-r-v01"
17
  tokenizer = AutoTokenizer.from_pretrained(model_id)