File size: 607 Bytes
b808313
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
#!/usr/bin/env python3
import torch
from modeling_rootformer import UnifiedRootformerV12
from morphemic_tokenizer import PureArabicMorphemicTokenizerV12

def main():
    print("Loading Rootformer v12 from local HuggingFace repo...")
    tok = PureArabicMorphemicTokenizerV12('rootformer_v12_arabic_blueprint.json')
    model = UnifiedRootformerV12(
        blueprint_path='rootformer_v12_arabic_blueprint.json',
        device='cuda' if torch.cuda.is_available() else 'cpu'
    )
    print("Rootformer v12 successfully instantiated with Farāhīdian radical layers!")

if __name__ == '__main__':
    main()