File size: 644 Bytes
2bbf97a
3e7c541
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
import torch
from transformers import AutoTokenizer, AutoModel

# Modell & Tokenizer laden
model_name = "allenai/scibert_scivocab_uncased"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModel.from_pretrained(model_name)

# Explizit auf CPU setzen
device = torch.device("cpu")
model.to(device)

# Beispiel-Text für den Test
text = "This paper introduces a novel deep learning approach for cancer diagnosis."

# Tokenisierung & Modell-Durchlauf
inputs = tokenizer(text, return_tensors="pt").to(device)
outputs = model(**inputs)

print("Modell erfolgreich geladen!")
print("Shape der Ausgabe:", outputs.last_hidden_state.shape)