groupuser/mistral-1L-tiny
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update 1/model.py

Browse Source
This commit is contained in:
cheetahadmin 2025-08-26 02:01:06 +00:00
parent b8398fa303
commit 0bb2560057
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
1/model.py
View File

@ -3,6 +3,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig
from peft import PeftModel, PeftConfig
import numpy as np
import json
import torch
class TritonPythonModel:
def initialize(self, args):