[fix] pr113

pull/115/head
jingyaogong 2026-04-01 15:40:25 +08:00
parent ac4c506eeb
commit 48adb0fda0
2 changed files with 1 additions and 3 deletions

View File

@ -63,8 +63,7 @@ def main():
inputs = tokenizer(inputs_text, return_tensors="pt", truncation=True).to(args.device) inputs = tokenizer(inputs_text, return_tensors="pt", truncation=True).to(args.device)
print(f'[图像]: {image_file}') print(f'[图像]: {image_file}')
display_prompt = prompt.replace('\n', '\\n') print(f"💬: {repr(prompt)}")
print(f'💬: {display_prompt}')
print('🤖: ', end='') print('🤖: ', end='')
st = time.time() st = time.time()
generated_ids = model.generate( generated_ids = model.generate(

@ -1 +0,0 @@
Subproject commit 9327b4fe10ef8352fa3a32eddc8d3c8bff984286