Hugging Face's logo

Spaces:
Yoxas
/
testchatbot
Runtime error

App Files Community
Yoxas commited on
Commit
9962a02
·
verified ·
1 Parent(s): 00ade96

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import json
7
  from transformers import AutoTokenizer, AutoModelForCausalLM, AutoModelForSequenceClassification
8
  import spaces
9
 
10
- #CUDA_LAUNCH_BLOCKING=1
11
 
12
  # Ensure you have GPU support
13
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
7
  from transformers import AutoTokenizer, AutoModelForCausalLM, AutoModelForSequenceClassification
8
  import spaces
9
 
10
+ CUDA_LAUNCH_BLOCKING=1
11
 
12
  # Ensure you have GPU support
13
  device = 'cuda' if torch.cuda.is_available() else 'cpu'