Hugging Face's logo

Spaces:
ntt123
/
diffusion-speech-360h
Running on Zero

App Files Community
ntt123 commited on
Commit
27914bf
·
1 Parent(s): b8b08c2

rm unused import

Browse files
Files changed (1) hide show
  1. models.py +1 -1
models.py CHANGED
@@ -15,7 +15,7 @@ import numpy as np
15
  import torch
16
  import torch.nn as nn
17
  import torch.nn.functional as F
18
- from torch.nn.attention.flex_attention import BlockMask, flex_attention
19
 
20
 
21
  class Mlp(nn.Module):
 
15
  import torch
16
  import torch.nn as nn
17
  import torch.nn.functional as F
18
+ # from torch.nn.attention.flex_attention import BlockMask, flex_attention
19
 
20
 
21
  class Mlp(nn.Module):