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
|