Skip to content

Commit

Permalink
Fix pytorch backend check in attention
Browse files Browse the repository at this point in the history
  • Loading branch information
casblaauw committed Jan 16, 2025
1 parent 2973d0d commit 8ca2701
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/crested/tl/zoo/utils/_attention.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
backend = keras.src.backend.config.backend()
if backend == 'tensorflow':
from tensorflow.math import lgamma
elif backend == 'pytorch':
elif backend == 'torch':
from torch import lgamma
else:
raise NotImplementedError(f"Using gamma position functions (as part of relative_position_functions == 'enformer') currently only supports TensorFlow and PyTorch backends, not {backend}.")
Expand Down

0 comments on commit 8ca2701

Please sign in to comment.