Skip to content

GitLab

Commit 16401cf5 authored by zhuww's avatar zhuww
Browse files

set _triton_available False

parent c785b534
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
fastfold/model/fastnn/kernel/layer_norm.py
View file @ 16401cf5
...@@ -4,7 +4,7 @@ import logging ...@@ -4,7 +4,7 @@ import logging
import torch import torch
from torch.nn.parameter import Parameter from torch.nn.parameter import Parameter
_triton_available = True _triton_available = False
if _triton_available: if _triton_available:
try: try:
from .triton.layer_norm import LayerNormTritonFunc from .triton.layer_norm import LayerNormTritonFunc
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment