MCPcopy
hub / github.com/zai-org/VisualGLM-6B / generate_input

Function generate_input

model/infer_util.py:35–43  ·  view source on GitHub ↗
(input_text, input_image_prompt, history=[], input_para=None, image_is_encoded=True)

Source from the content-addressed store, hash-verified

33 return zh_pattern.search(text)
34
35def generate_input(input_text, input_image_prompt, history=[], input_para=None, image_is_encoded=True):
36 if not image_is_encoded:
37 image = input_image_prompt
38 else:
39 decoded_image = base64.b64decode(input_image_prompt)
40 image = Image.open(BytesIO(decoded_image))
41
42 input_data = {'input_query': input_text, 'input_image': image, 'history': history, 'gen_kwargs': input_para}
43 return input_data
44
45
46def process_image(image_encoded):

Callers 2

visual_glmFunction · 0.90
generate_text_with_imageFunction · 0.90

Calls

no outgoing calls

Tested by

no test coverage detected