pushed from different env

neulab · Oct 11, 2024 · 2a7c529 · 2a7c529
1 parent 096c683
commit 2a7c529
Showing 1 changed file with 3 additions and 15 deletions.
diff --git a/examples/community_lm/community_lm_rag.ipynb b/examples/community_lm/community_lm_rag.ipynb
@@ -32,10 +32,10 @@
     "from llments.datastore.pyserini_datastore import PyseriniDatastore\n",
     "import pandas as pd\n",
     "import numpy as np\n",
-    "from community_lm_constants import politician_feelings, groups_feelings, anes_df\n",
-    "from community_lm_utils import generate_community_opinion, compute_group_stance, generate_community_opinion_rag\n",
+    "from examples.community_lm.community_lm_constants import politician_feelings, groups_feelings, anes_df\n",
+    "from examples.community_lm.community_lm_utils import generate_community_opinion, compute_group_stance, generate_community_opinion_rag\n",
     "\n",
-    "device = 'cuda'  # change to 'mps' if you have a mac, or 'cuda:0' if you have an NVIDIA GPU "
+    "device = 'cuda'  # change to 'mps' if you have a mac, or 'cuda' if you have an NVIDIA GPU "
    ]
   },
   {
@@ -4760,18 +4760,6 @@
    "display_name": "Python 3 (ipykernel)",
    "language": "python",
    "name": "python3"
-  },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.11.8"
   }
  },
  "nbformat": 4,