Skip to content

Commit

Permalink
Update LLM_FT.py
Browse files Browse the repository at this point in the history
  • Loading branch information
igorgonteri authored Nov 15, 2024
1 parent 4db8065 commit 049b284
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions 0007-LLM-FT/LLM_FT.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
from datasets import Dataset
from transformers import AutoModelForCausalLM

hf_token = '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'
hf_token =
login(token=hf_token, add_to_git_credential=True)

def open_file(filepath):
Expand Down Expand Up @@ -74,7 +74,7 @@ def to_sequence(self, peptide_str: str) -> str:

model = AutoModelForCausalLM.from_pretrained('meta-llama/Llama-3.2-1B')

model_checkpoint = "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"
model_checkpoint =
tokenizer = AutoTokenizer.from_pretrained(model_checkpoint, token=hf_token)

dataset_path = '/Users/igorgonteri/Desktop/LLM_fine_tuning/re_datasets/RW_lexicon.txt'
Expand Down

0 comments on commit 049b284

Please sign in to comment.