GlandVergil commited on
Commit
ff2d515
1 Parent(s): 2e2a6fa

Update rfdiffusion/util.py

Browse files
Files changed (1) hide show
  1. 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)