Spaces:

Pipatpong
/

VCM_Demo

Runtime error

Pipatpong commited on Jun 6, 2023

Commit

45edbb8

1 Parent(s): 716692e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,6 +4,7 @@ import gradio as gr
 import re
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 checkpoint = "Pipatpong/vcm_santa"
 device = "cuda" if torch.cuda.is_available() else "CPU"

 import re
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
+from bitsandbytes import BitsAndBytesConfig
 checkpoint = "Pipatpong/vcm_santa"
 device = "cuda" if torch.cuda.is_available() else "CPU"