bartduis commited on
Commit
744eeee
·
1 Parent(s): fa9cd0b
Files changed (2) hide show
  1. models/pos_embed.py +2 -1
  2. requirements.txt +0 -1
models/pos_embed.py CHANGED
@@ -105,7 +105,8 @@ try:
105
  RoPE2D = cuRoPE2D
106
  except ImportError:
107
  # critical error, we need to use the slow pytorch version
108
- raise ImportError("CUDA-compiled version of RoPE2D is required but could not be found. Please compile the CUDA extension before running.")
 
109
 
110
  class RoPE2D(torch.nn.Module):
111
 
 
105
  RoPE2D = cuRoPE2D
106
  except ImportError:
107
  # critical error, we need to use the slow pytorch version
108
+ print("CUDA-compiled version of RoPE2D is required but could not be found. Please compile the CUDA extension before running.")
109
+ #raise ImportError("CUDA-compiled version of RoPE2D is required but could not be found. Please compile the CUDA extension before running.")
110
 
111
  class RoPE2D(torch.nn.Module):
112
 
requirements.txt CHANGED
@@ -1,4 +1,3 @@
1
- git+https://github.com/Duisterhof/rayst3r.git#egg=curope&subdirectory=extensions/curope
2
  matplotlib
3
  numpy
4
  open3d
 
 
1
  matplotlib
2
  numpy
3
  open3d