Hugging Face's logo

Spaces:
liuganghuggingface
/
Polymer-Design-With-GraphDiT
Runtime error

App Files Community
liuganghuggingface commited on
Commit
c373e8c
·
verified ·
1 Parent(s): 91ba725

Update graph_decoder/transformer.py

Browse files
Files changed (1) hide show
  1. graph_decoder/transformer.py +2 -0
graph_decoder/transformer.py CHANGED
@@ -1,3 +1,5 @@
 
 
1
  import torch
2
  import torch.nn as nn
3
  from .layers import Attention, MLP
 
1
+ import spaces
2
+
3
  import torch
4
  import torch.nn as nn
5
  from .layers import Attention, MLP