mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-16 11:08:23 +06:00
1c4b62b219
* some config changes * update * current state * update * update * updates and cleanup * something that works * fixup * fixes * nits * nit * nits and fix * Update src/transformers/integrations/tensor_parallel.py Co-authored-by: Lysandre Debut <hi@lysand.re> * Update src/transformers/integrations/tensor_parallel.py Co-authored-by: Lysandre Debut <hi@lysand.re> * cleanup * style * safe import * fix * updates * rename stuff an clean * style * small updates * ups * oups * nit * protect imports * update tp * rodfl * arf * turbo nit on init * fix import error * frumble gumbgle * try to fix the import error * should fix the non model test * update keep in float32 * update * fix * nits * fix subvconfigs * test was weird * nit * fix failing test * fix instruct blip * fixes * style * x.com * fix overwrite * ok last bit of failing test --------- Co-authored-by: Lysandre Debut <hi@lysand.re>
71 lines
2.3 KiB
Python
71 lines
2.3 KiB
Python
import tempfile
|
|
import unittest
|
|
|
|
from transformers import LlavaConfig
|
|
|
|
|
|
class LlavaConfigTest(unittest.TestCase):
|
|
def test_llava_reload(self):
|
|
"""
|
|
Simple test for reloading default llava configs
|
|
"""
|
|
with tempfile.TemporaryDirectory() as tmp_dir:
|
|
config = LlavaConfig()
|
|
config.save_pretrained(tmp_dir)
|
|
|
|
reloaded = LlavaConfig.from_pretrained(tmp_dir)
|
|
assert config.to_dict() == reloaded.to_dict()
|
|
|
|
def test_pixtral_reload(self):
|
|
"""
|
|
Simple test for reloading pixtral configs
|
|
"""
|
|
vision_config = {
|
|
"model_type": "pixtral",
|
|
"head_dim": 64,
|
|
"hidden_act": "silu",
|
|
"image_size": 1024,
|
|
"is_composition": True,
|
|
"patch_size": 16,
|
|
"rope_theta": 10000.0,
|
|
"tie_word_embeddings": False,
|
|
}
|
|
|
|
text_config = {
|
|
"model_type": "mistral",
|
|
"hidden_size": 5120,
|
|
"head_dim": 128,
|
|
"num_attention_heads": 32,
|
|
"intermediate_size": 14336,
|
|
"is_composition": True,
|
|
"max_position_embeddings": 1024000,
|
|
"num_hidden_layers": 40,
|
|
"num_key_value_heads": 8,
|
|
"rms_norm_eps": 1e-05,
|
|
"rope_theta": 1000000000.0,
|
|
"sliding_window": None,
|
|
"vocab_size": 131072,
|
|
}
|
|
|
|
with tempfile.TemporaryDirectory() as tmp_dir:
|
|
config = LlavaConfig(vision_config=vision_config, text_config=text_config)
|
|
config.save_pretrained(tmp_dir)
|
|
|
|
reloaded = LlavaConfig.from_pretrained(tmp_dir)
|
|
assert config.to_dict() == reloaded.to_dict()
|
|
|
|
def test_arbitrary_reload(self):
|
|
"""
|
|
Simple test for reloading arbirarily composed subconfigs
|
|
"""
|
|
default_values = LlavaConfig().to_diff_dict()
|
|
default_values["vision_config"]["model_type"] = "pixtral"
|
|
default_values["text_config"]["model_type"] = "opt"
|
|
self.maxDiff = None
|
|
with tempfile.TemporaryDirectory() as tmp_dir:
|
|
config = LlavaConfig(**default_values)
|
|
config.save_pretrained(tmp_dir)
|
|
|
|
reloaded = LlavaConfig.from_pretrained(tmp_dir)
|
|
self.assertDictEqual(config.to_dict(), reloaded.to_dict())
|