flytoe commited on
Commit
2bbf97a
·
verified ·
1 Parent(s): def161a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -6
app.py CHANGED
@@ -1,6 +1,3 @@
1
- from transformers import AutoTokenizer, AutoModel
2
-
3
- # Explizite Angabe von torch_dtype kann helfen
4
- model_name = "allenai/scibert_scivocab_uncased"
5
- tokenizer = AutoTokenizer.from_pretrained(model_name)
6
- model = AutoModel.from_pretrained(model_name, torch_dtype=torch.float32)
 
1
+ import torch
2
+ print(torch.__version__)
3
+ print(torch.cuda.is_available()) # Zeigt, ob eine GPU erkannt wird