Instructions to use igorktech/hierarchical-dialog-bert with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use igorktech/hierarchical-dialog-bert with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("fill-mask", model="igorktech/hierarchical-dialog-bert", trust_remote_code=True)# Load model directly from transformers import AutoModelForMaskedLM model = AutoModelForMaskedLM.from_pretrained("igorktech/hierarchical-dialog-bert", trust_remote_code=True, dtype="auto") - Notebooks
- Google Colab
- Kaggle
Update modelling_hat.py
Browse files- modelling_hat.py +1 -1
modelling_hat.py
CHANGED
|
@@ -431,7 +431,7 @@ class SentenceClassifierOutput(ModelOutput):
|
|
| 431 |
|
| 432 |
|
| 433 |
|
| 434 |
-
from configuration_hat import HATConfig
|
| 435 |
|
| 436 |
class HATEmbeddings(nn.Module):
|
| 437 |
"""
|
|
|
|
| 431 |
|
| 432 |
|
| 433 |
|
| 434 |
+
from .configuration_hat import HATConfig
|
| 435 |
|
| 436 |
class HATEmbeddings(nn.Module):
|
| 437 |
"""
|