Skip to content

Commit

Permalink
should stop instantiation of java
Browse files Browse the repository at this point in the history
  • Loading branch information
Parry-Parry committed Nov 6, 2024
1 parent fdb5cd8 commit e58f29b
Show file tree
Hide file tree
Showing 3 changed files with 0 additions and 6 deletions.
2 changes: 0 additions & 2 deletions rankers/modelling/cat.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,4 @@
import pyterrier as pt
if not pt.started():
pt.init()
from transformers import PreTrainedModel, PretrainedConfig, PreTrainedTokenizer, AutoModelForSequenceClassification, AutoTokenizer, AutoConfig
from typing import Union
import torch
Expand Down
2 changes: 0 additions & 2 deletions rankers/modelling/dot.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,8 +3,6 @@
import torch
from torch import nn
import pyterrier as pt
if not pt.started():
pt.init()
from transformers import PreTrainedModel, PreTrainedTokenizer, PretrainedConfig, AutoModel, AutoTokenizer, AutoConfig
from typing import Union
import pandas as pd
Expand Down
2 changes: 0 additions & 2 deletions rankers/modelling/seq2seq.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,4 @@
import pyterrier as pt
if not pt.started():
pt.init()
from transformers import PreTrainedModel, PreTrainedConfig, PreTrainedTokenizer, AutoTokenizer, AutoConfig, AutoModelForSeq2SeqLM, AutoModelForCausalLM
from typing import Union
import torch
Expand Down

0 comments on commit e58f29b

Please sign in to comment.