From da990bc2255d8ed5099e851a8e22d2b4b1d53bba Mon Sep 17 00:00:00 2001
From: Artem Chumachenko <artek.chumak@gmail.com>
Date: Tue, 5 Mar 2024 13:45:16 +0400
Subject: [PATCH] Fix versions

---
 src/petals/utils/peft.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/petals/utils/peft.py b/src/petals/utils/peft.py
index 510f90e..a47bf79 100644
--- a/src/petals/utils/peft.py
+++ b/src/petals/utils/peft.py
@@ -1,7 +1,7 @@
 import contextlib
 import re
 import time
-from typing import Optional, Sequence, Union
+from typing import List, Optional, Sequence, Union
 
 import bitsandbytes as bnb
 import torch
@@ -157,7 +157,7 @@ class AdapterContextMixin:
     def active_adapters(self):
         return [self._context_active_adapter]
 
-    def set_adapter(self, adapter_names: str | list[str]) -> None:
+    def set_adapter(self, adapter_names) -> None:
         """
         In PEFT, this function making adapter trainable. However, in Petals environment is not possible now. So,
         this code remove this functionality.