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 from peft import PeftModel, PeftConfig
import numpy as np import numpy as np
import json import json
import torch
class TritonPythonModel: class TritonPythonModel:
def initialize(self, args): def initialize(self, args):