lambdaofgod commited on
Commit
655f181
β€’
1 Parent(s): c5a2694

cuda bug, file renames

Browse files
README.md CHANGED
@@ -5,7 +5,7 @@ colorFrom: red
5
  colorTo: gray
6
  sdk: streamlit
7
  sdk_version: 1.15.2
8
- app_file: project_retrieval_app.py
9
  pinned: false
10
  ---
11
 
 
5
  colorTo: gray
6
  sdk: streamlit
7
  sdk_version: 1.15.2
8
+ app_file: app.py
9
  pinned: false
10
  ---
11
 
project_retrieval_app.py β†’ app.py RENAMED
File without changes
app_implementation.py CHANGED
@@ -16,13 +16,13 @@ from search_utils import (
16
 
17
 
18
  class RetrievalApp:
 
19
  def is_cuda_available(self):
20
  try:
21
- t = torch.Tensor([1]).cuda()
22
  except:
23
  return False
24
- finally:
25
- return True
26
 
27
  def get_device_options(self):
28
  if self.is_cuda_available():
 
16
 
17
 
18
  class RetrievalApp:
19
+
20
  def is_cuda_available(self):
21
  try:
22
+ torch._C._cuda_init()
23
  except:
24
  return False
25
+ return True
 
26
 
27
  def get_device_options(self):
28
  if self.is_cuda_available():
pages/{2_Statistics.py β†’ 2_Metrics.py} RENAMED
File without changes