How to use VatsalPatel18/EdgeWeightPredictor with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("VatsalPatel18/EdgeWeightPredictor", dtype="auto")
