commit
ac4c506eeb
|
|
@ -63,7 +63,8 @@ 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}')
|
||||||
print(f'💬: {prompt.replace('\n', '\\n')}')
|
display_prompt = prompt.replace('\n', '\\n')
|
||||||
|
print(f'💬: {display_prompt}')
|
||||||
print('🤖: ', end='')
|
print('🤖: ', end='')
|
||||||
st = time.time()
|
st = time.time()
|
||||||
generated_ids = model.generate(
|
generated_ids = model.generate(
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1 @@
|
||||||
|
Subproject commit 9327b4fe10ef8352fa3a32eddc8d3c8bff984286
|
||||||
Loading…
Reference in New Issue