Skip to content
Snippets Groups Projects
Commit 20dd4740 authored by Himanshu Shukla's avatar Himanshu Shukla
Browse files

added working code of CLI/gradio UI/ LoRA weights merge

parent 24333253
No related branches found
No related tags found
No related merge requests found
......@@ -16,7 +16,7 @@ DEFAULT_MODEL = "meta-llama/Llama-3.2-11B-Vision-Instruct"
def load_model_and_processor(model_name: str, hf_token: str, finetuning_path: str = None):
"""
Load the model and processor, and optionally load adapter weights if specified.
Load the model and processor, and optionally load adapter weights if specified
"""
# Load pre-trained model and processor
model = MllamaForConditionalGeneration.from_pretrained(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment