POKASH.PL
/
ably.do
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

test

This commit is contained in:
l.gabrysiak 2025-02-26 10:22:13 +01:00
parent 97e9bc647f
commit 8ee5f5cbd9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test.py
View File

@ -6,7 +6,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_path)
tokenizer.pad_token = tokenizer.eos_token tokenizer.pad_token = tokenizer.eos_token
model.config.pad_token_id = tokenizer.eos_token_id model.config.pad_token_id = tokenizer.eos_token_id
def generate_response(prompt, max_length=100): def generate_response(prompt, max_length=1000):
inputs = tokenizer(prompt, return_tensors="pt", padding=True, truncation=True) inputs = tokenizer(prompt, return_tensors="pt", padding=True, truncation=True)
outputs = model.generate( outputs = model.generate(
inputs.input_ids, inputs.input_ids,