Ucas-HaoranWei/Vary

运行demo报错

Opened this issue · 1 comments

Traceback (most recent call last):
File "/root/lh/Vary-main/Vary-master/vary/demo/run_qwen_vary.py", line 8, in
from vary.model import *
File "/root/miniconda3/envs/vary/lib/python3.10/site-packages/vary/model/init.py", line 1, in
from .vary_opt import varyOPTModel, varyOPTForCausalLM
File "/root/miniconda3/envs/vary/lib/python3.10/site-packages/vary/model/vary_opt.py", line 37, in
from vary.model.plug.transforms import train_transform, test_transform
File "/root/miniconda3/envs/vary/lib/python3.10/site-packages/vary/model/plug/transforms.py", line 146, in
alb.ElasticTransform(
File "/root/miniconda3/envs/vary/lib/python3.10/site-packages/albumentations/core/validation.py", line 35, in custom_init
config = dct"InitSchema"
File "/root/miniconda3/envs/vary/lib/python3.10/site-packages/pydantic/main.py", line 193, in init
self.pydantic_validator.validate_python(data, self_instance=self)
pydantic_core._pydantic_core.ValidationError: 1 validation error for InitSchema
alpha_affine
Input should be None [type=none_required, input_value=1.2, input_type=float]
For further information visit https://errors.pydantic.dev/2.8/v/none_required

transforms.py中alpha_affine = None