update generato

generator.ipynb

@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": null,
    "id": "bae751d8",
    "metadata": {},
    "outputs": [],
@@ -15,12 +15,12 @@
     "from tqdm import tqdm\n",
     "\n",
     "from torch.utils.data import DataLoader\n",
-    "from models.open_mp_gen.tokenizer import Tokenizer\n",
-    "from models.open_mp_gen.model.generator import Generator\n",
-    "from models.open_mp_gen.model.encoder import Encoder\n",
-    "from models.open_mp_gen.model.decoder import Decoder\n",
-    "from models.open_mp_gen.model.attn import BahdanauAttention\n",
-    "from models.open_mp_gen.dataset import OpenMPDataset\n",
+    "from tokenizer import Tokenizer\n",
+    "from model.generator import Generator\n",
+    "from model.encoder import Encoder\n",
+    "from model.decoder import Decoder\n",
+    "from model.attn import BahdanauAttention\n",
+    "from dataset import OpenMPDataset\n",
     "from accelera.src.utils.code_utils import pragma_to_class"
    ]
   },