Former-commit-id: 8edb73116b3be54a5a278dd32bbd6819cafeec92
pull/160/head
Kye 2 years ago
parent a2cfc2b6ed
commit 6114a86855

@ -2,7 +2,7 @@ import pytest
import torch
from unittest.mock import Mock, MagicMock
from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
from your_module import HuggingFaceLLM # replace with actual import
from swarms.agents.models.hf import HuggingFaceLLM # replace with actual import
@pytest.fixture

Loading…
Cancel
Save