commit to moving the file to models entirely now that everyone elses

changes are in
nod-ai · Nov 13, 2023 · 0a475d5 · 0a475d5
1 parent 856c559
commit 0a475d5
Show file tree

Hide file tree

Showing 5 changed files with 40 additions and 393 deletions.
diff --git a/examples/llama2_inference/llama2.ipynb b/examples/llama2_inference/llama2.ipynb
@@ -7,19 +7,11 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import os\n",
-    "import numpy as np\n",
-    "from shark_turbine.dynamo.passes import turbine_cpu_pass_pipeline\n",
-    "from torch._export import dynamic_dim\n",
-    "from torch._export.constraints import constrain_as_size, constrain_as_value\n",
     "from transformers import AutoTokenizer, AutoModelForCausalLM\n",
     "import torch\n",
     "from torch.utils import _pytree as pytree\n",
-    "from torch.fx import GraphModule\n",
-    "from torch._export.constraints import constrain_as_size, constrain_as_value\n",
     "from shark_turbine.aot import *\n",
-    "from iree.compiler.ir import Context\n",
-    "from iree import runtime as ireert\n"
+    "from iree.compiler.ir import Context\n"
    ]
   },
   {

diff --git a/python/turbine_models/custom_models/llama2.py b/python/turbine_models/custom_models/llama2.py
diff --git a/examples/llama2_inference/remap_gguf.py → ...urbine_models/custom_models/remap_gguf.py b/examples/llama2_inference/remap_gguf.py → ...urbine_models/custom_models/remap_gguf.py