Added missing import

Author: rosinality

convert_weight.py

@@ -198,7 +198,7 @@ def fill_statedict(state_dict, vars, size):
 
 
 if __name__ == "__main__":
-    device = "cuda"
+    device = "cpu"
 
     parser = argparse.ArgumentParser()
     parser.add_argument("--repo", type=str, required=True)

op/fused_act.py

@@ -2,6 +2,7 @@
 
 import torch
 from torch import nn
+from torch.nn import functional as F
 from torch.autograd import Function
 from torch.utils.cpp_extension import load