Spaces:
Running
on
T4
Running
on
T4
Update ROBERTAmodel.py
Browse files- ROBERTAmodel.py +3 -3
ROBERTAmodel.py
CHANGED
|
@@ -35,7 +35,7 @@ class RoBERTaVisualizer(TransformerVisualizer):
|
|
| 35 |
MODEL = "roberta-base"
|
| 36 |
LOCAL_PATH = os.path.join(CACHE_DIR, "models",MODEL)
|
| 37 |
|
| 38 |
-
self.model = RobertaForMaskedLM.from_pretrained( LOCAL_PATH, local_files_only=True )
|
| 39 |
"""
|
| 40 |
try:
|
| 41 |
self.model = RobertaForMaskedLM.from_pretrained( LOCAL_PATH, local_files_only=True )
|
|
@@ -49,7 +49,7 @@ class RoBERTaVisualizer(TransformerVisualizer):
|
|
| 49 |
MODEL = 'textattack_roberta-base-SST-2'
|
| 50 |
LOCAL_PATH = os.path.join(CACHE_DIR, "models",MODEL)
|
| 51 |
|
| 52 |
-
self.model = RobertaForSequenceClassification.from_pretrained( LOCAL_PATH, local_files_only=True )
|
| 53 |
"""
|
| 54 |
try:
|
| 55 |
self.model = RobertaForSequenceClassification.from_pretrained( LOCAL_PATH, local_files_only=True )
|
|
@@ -63,7 +63,7 @@ class RoBERTaVisualizer(TransformerVisualizer):
|
|
| 63 |
LOCAL_PATH = os.path.join(CACHE_DIR, "models",MODEL)
|
| 64 |
|
| 65 |
|
| 66 |
-
self.model = RobertaForSequenceClassification.from_pretrained( LOCAL_PATH, local_files_only=True)
|
| 67 |
"""
|
| 68 |
try:
|
| 69 |
self.model = RobertaForSequenceClassification.from_pretrained( LOCAL_PATH, local_files_only=True)
|
|
|
|
| 35 |
MODEL = "roberta-base"
|
| 36 |
LOCAL_PATH = os.path.join(CACHE_DIR, "models",MODEL)
|
| 37 |
|
| 38 |
+
self.model = RobertaForMaskedLM.from_pretrained( LOCAL_PATH, local_files_only=True ).to(self.device)
|
| 39 |
"""
|
| 40 |
try:
|
| 41 |
self.model = RobertaForMaskedLM.from_pretrained( LOCAL_PATH, local_files_only=True )
|
|
|
|
| 49 |
MODEL = 'textattack_roberta-base-SST-2'
|
| 50 |
LOCAL_PATH = os.path.join(CACHE_DIR, "models",MODEL)
|
| 51 |
|
| 52 |
+
self.model = RobertaForSequenceClassification.from_pretrained( LOCAL_PATH, local_files_only=True ).to(self.device)
|
| 53 |
"""
|
| 54 |
try:
|
| 55 |
self.model = RobertaForSequenceClassification.from_pretrained( LOCAL_PATH, local_files_only=True )
|
|
|
|
| 63 |
LOCAL_PATH = os.path.join(CACHE_DIR, "models",MODEL)
|
| 64 |
|
| 65 |
|
| 66 |
+
self.model = RobertaForSequenceClassification.from_pretrained( LOCAL_PATH, local_files_only=True).to(self.device)
|
| 67 |
"""
|
| 68 |
try:
|
| 69 |
self.model = RobertaForSequenceClassification.from_pretrained( LOCAL_PATH, local_files_only=True)
|