Hugging Face's logo

Spaces:
Pipatpong
/
VCM_Demo
Runtime error

App Files Community
Pipatpong commited on
Commit
b0400a5
·
1 Parent(s): 45edbb8

Update config

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -3,8 +3,7 @@
3
  import gradio as gr
4
  import re
5
  import torch
6
- from transformers import AutoModelForCausalLM, AutoTokenizer
7
- from bitsandbytes import BitsAndBytesConfig
8
 
9
  checkpoint = "Pipatpong/vcm_santa"
10
  device = "cuda" if torch.cuda.is_available() else "CPU"
 
3
  import gradio as gr
4
  import re
5
  import torch
6
+ from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
 
7
 
8
  checkpoint = "Pipatpong/vcm_santa"
9
  device = "cuda" if torch.cuda.is_available() else "CPU"