()
| 32 | |
| 33 | |
| 34 | def test_load_model_transformers_model(): |
| 35 | with ( |
| 36 | patch( |
| 37 | "transformers.AutoModelForImageTextToText.from_pretrained", |
| 38 | side_effect=ValueError("Unrecognized configuration class"), |
| 39 | ), |
| 40 | patch("transformers.AutoModelForCausalLM.from_pretrained"), |
| 41 | patch("transformers.AutoTokenizer.from_pretrained"), |
| 42 | ): |
| 43 | model = load_model("TransformersModel", "test_model_id") |
| 44 | assert isinstance(model, TransformersModel) |
| 45 | assert model.model_id == "test_model_id" |
| 46 | |
| 47 | |
| 48 | def test_load_model_hf_api_model(set_env_vars): |
nothing calls this directly
no test coverage detected
searching dependent graphs…