bp
Browse files- models/pos_embed.py +2 -1
- 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 |
-
|
|
|
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
|