mirror of
https://github.com/lucasjinreal/Namo-R1.git
synced 2026-01-13 22:07:17 +08:00
20 lines
272 B
Python
20 lines
272 B
Python
import torch
|
|
|
|
|
|
def load(pretrained_model_path):
|
|
"""
|
|
returns model, tokenizer, image_processor
|
|
"""
|
|
pass
|
|
|
|
|
|
def generate(model, tokenizer, prompt, images):
|
|
pass
|
|
|
|
|
|
if __name__ == "__main__":
|
|
model = load("")
|
|
output = generate(
|
|
model,
|
|
)
|