replace get_default_device with PartialState

requirements.txt

@@ -1,4 +1,4 @@
-transformers>=4.29.0
+transformers>=4.35.2
 diffusers
 accelerate
 torch

text_to_image.py

@@ -1,4 +1,4 @@
-from transformers.tools.base import Tool, get_default_device
+from transformers.tools.base import Tool
 from transformers.utils import is_accelerate_available
 import torch
 
@@ -18,7 +18,9 @@ class TextToImageTool(Tool):
     outputs = ['image']
 
     def __init__(self, device=None, **hub_kwargs) -> None:
-        if not is_accelerate_available():
+        if is_accelerate_available():
+            from accelerate import PartialState
+        else:
             raise ImportError("Accelerate should be installed in order to use tools.")
 
         super().__init__()
@@ -29,7 +31,7 @@ class TextToImageTool(Tool):
 
     def setup(self):
         if self.device is None:
-            self.device = get_default_device()
+            self.device = PartialState().default_device
 
         self.pipeline = DiffusionPipeline.from_pretrained(self.default_checkpoint)
         self.pipeline.scheduler = DPMSolverMultistepScheduler.from_config(self.pipeline.scheduler.config)