Spaces:

david-thrower
/

3B-Param-Basic-Chatbot

Build error

david-thrower commited on 18 days ago

Commit

2013b5a

verified ·

1 Parent(s): bd1d5de

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,6 +1,7 @@
 import gradio as gr
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 MODEL_ID = "HuggingFaceTB/SmolLM3-3B"
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"

 import gradio as gr
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
+import bitsandbytes
 MODEL_ID = "HuggingFaceTB/SmolLM3-3B"
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"