Hugging Face's logo

medmekk
/
rmsnorm

kernel
Model card Files Community
medmekk HF Staff commited on
Commit
c9d5520
·
verified ·
1 Parent(s): bb342db

Update build-output/torch-universal/rmsnorm/rmsnorm.py

Browse files
Files changed (1) hide show
  1. build-output/torch-universal/rmsnorm/rmsnorm.py +0 -1
build-output/torch-universal/rmsnorm/rmsnorm.py CHANGED
@@ -10,7 +10,6 @@ import math
10
 
11
  import torch
12
  import torch.nn.functional as F
13
- from torch.amp import custom_fwd, custom_bwd
14
 
15
  import triton
16
  import triton.language as tl
 
10
 
11
  import torch
12
  import torch.nn.functional as F
 
13
 
14
  import triton
15
  import triton.language as tl