Billy Cao
Billy Cao
- Add support for batched inference on slices - Add support for TensorRT for Ultralytics-based/originally YOLOv8 models (exported using Ultralytics) - Add support for YOLOv6 (requires [yolov6](https://github.com/meituan/YOLOv6) repo on sys.path)...
### Feature request Add support for export SigLIP models ### Motivation As used by many SOTA VLMs, SigLIP is gaining traction and supporting it can be the step 1 to...
Because how flattener limits to floats as inputs, it will wrongly remove any int inputs that are common in text models e.g. encoded token ids. Discovered when trying to convert...
``` import torch from transformers import AutoImageProcessor, AutoModelForZeroShotImageClassification, AutoTokenizer, ZeroShotImageClassificationPipeline, SiglipProcessor, SiglipModel from torch2trt import torch2trt model = SiglipModel.from_pretrained('google/siglip-large-patch16-384', torch_dtype=torch.float16).cuda() text_model = model.text_model dummy = torch.ones(1, 64, dtype=torch.long, device='cuda') text_model(dummy)...
Allow user to pass in a list of np arrays to `enhance` method like this: ```python from basicsr.archs.rrdbnet_arch import RRDBNet from realesrgan import RealESRGANer from realesrgan.archs.srvgg_arch import SRVGGNetCompact model =...
# What does this PR do? Fixes #5444 The current impl disregards whatever user might have set in the yaml config during evaluations that happen during training (as specified by...
### Reminder - [X] I have read the README and searched the existing issues. ### System Info - `llamafactory` version: 0.9.1.dev0 - Platform: Linux-6.5.0-27-generic-x86_64-with-glibc2.35 - Python version: 3.10.12 - PyTorch...