[fix] pr113
parent
ac4c506eeb
commit
48adb0fda0
|
|
@ -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
|
|
||||||
Loading…
Reference in New Issue