commit
ac4c506eeb
|
|
@ -63,7 +63,8 @@ def main():
|
|||
inputs = tokenizer(inputs_text, return_tensors="pt", truncation=True).to(args.device)
|
||||
|
||||
print(f'[图像]: {image_file}')
|
||||
print(f'💬: {prompt.replace('\n', '\\n')}')
|
||||
display_prompt = prompt.replace('\n', '\\n')
|
||||
print(f'💬: {display_prompt}')
|
||||
print('🤖: ', end='')
|
||||
st = time.time()
|
||||
generated_ids = model.generate(
|
||||
|
|
|
|||
|
|
@ -0,0 +1 @@
|
|||
Subproject commit 9327b4fe10ef8352fa3a32eddc8d3c8bff984286
|
||||
Loading…
Reference in New Issue