Spaces:
Runtime error
Runtime error
GlandVergil
commited on
Commit
•
ff2d515
1
Parent(s):
2e2a6fa
Update rfdiffusion/util.py
Browse files- rfdiffusion/util.py +1 -1
rfdiffusion/util.py
CHANGED
@@ -250,7 +250,7 @@ def make_frame(X, Y):
|
|
250 |
Xn = X / torch.linalg.norm(X)
|
251 |
Y = Y - torch.dot(Y, Xn) * Xn
|
252 |
Yn = Y / torch.linalg.norm(Y)
|
253 |
-
Z = torch.cross(Xn, Yn)
|
254 |
Zn = Z / torch.linalg.norm(Z)
|
255 |
|
256 |
return torch.stack((Xn, Yn, Zn), dim=-1)
|
|
|
250 |
Xn = X / torch.linalg.norm(X)
|
251 |
Y = Y - torch.dot(Y, Xn) * Xn
|
252 |
Yn = Y / torch.linalg.norm(Y)
|
253 |
+
Z = torch.linalg.cross(Xn, Yn)
|
254 |
Zn = Z / torch.linalg.norm(Z)
|
255 |
|
256 |
return torch.stack((Xn, Yn, Zn), dim=-1)
|