Hugging Face's logo

Spaces:
facat
/
alpaca-lora-cn
Runtime error

App Files Community
facat commited on
Commit
9e51263
•
1 Parent(s): a299003

debug

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -1,4 +1,6 @@
1
  # %%
 
 
2
  import gradio as gr
3
  from transformers import LlamaTokenizer
4
  from transformers import LlamaForCausalLM, GenerationConfig
 
1
  # %%
2
+ import os
3
+ os.environ["CUDA_VISIBLE_DEVICES"] = ""
4
  import gradio as gr
5
  from transformers import LlamaTokenizer
6
  from transformers import LlamaForCausalLM, GenerationConfig