Skip to content

Commit

Permalink
try this change
Browse files Browse the repository at this point in the history
  • Loading branch information
mihir86 committed Oct 10, 2024
1 parent 5fab68e commit e40f0d7
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions examples/community_lm/community_lm_rag.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -32,8 +32,8 @@
"from llments.datastore.pyserini_datastore import PyseriniDatastore\n",
"import pandas as pd\n",
"import numpy as np\n",
"from 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",
"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",
"\n",
"device = 'cuda' # change to 'mps' if you have a mac, or 'cuda:0' if you have an NVIDIA GPU "
]
Expand Down

0 comments on commit e40f0d7

Please sign in to comment.