From d0a40969b275aec73db7e5d0357e140cdd74bf82 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=D0=9C=D0=B0=D0=B7=D1=83=D1=80=20=D0=93=D1=80=D0=B5=D1=82?=
=?UTF-8?q?=D0=B0=20=D0=95=D0=B2=D0=B3=D0=B5=D0=BD=D1=8C=D0=B5=D0=B2=D0=BD?=
=?UTF-8?q?=D0=B0?= <gemazur_1@edu.hse.ru>
Date: Tue, 25 Mar 2025 22:21:25 +0300
Subject: [PATCH] micro zapusk no cross
---
.ipynb_checkpoints/trytoubload-checkpoint.py | 2 ++
trytoubload.py | 2 ++
2 files changed, 4 insertions(+)
diff --git a/.ipynb_checkpoints/trytoubload-checkpoint.py b/.ipynb_checkpoints/trytoubload-checkpoint.py
index b483626..e04f14a 100644
--- a/.ipynb_checkpoints/trytoubload-checkpoint.py
+++ b/.ipynb_checkpoints/trytoubload-checkpoint.py
@@ -14,6 +14,8 @@ from transformers import BertTokenizer, BertPreTrainedModel, BertModel, Trainer,
from torch import nn
from peft import get_peft_model, LoraConfig, TaskType
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+
class MultiTaskBert(BertPreTrainedModel):
def __init__(self, config):
diff --git a/trytoubload.py b/trytoubload.py
index b483626..e04f14a 100644
--- a/trytoubload.py
+++ b/trytoubload.py
@@ -14,6 +14,8 @@ from transformers import BertTokenizer, BertPreTrainedModel, BertModel, Trainer,
from torch import nn
from peft import get_peft_model, LoraConfig, TaskType
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+
class MultiTaskBert(BertPreTrainedModel):
def __init__(self, config):
--
GitLab