update

asset3d_gen/validators/aesthetic_predictor.py

@@ -18,11 +18,9 @@ class AestheticPredictor:
         device (str): Device to use for computation ("cuda" or "cpu").
     """
 
-    def __init__(self, clip_model_dir=None, sac_model_path=None, device=None):
+    def __init__(self, clip_model_dir=None, sac_model_path=None, device="cpu"):
 
-        self.device = device or (
-            "cuda" if torch.cuda.is_available() else "cpu"
-        )
+        self.device = device
 
         if clip_model_dir is None:
             model_path = snapshot_download(
@@ -91,7 +89,6 @@ class AestheticPredictor:
         model.eval()
         return model
 
-    @spaces.GPU
     def predict(self, image_path):
         """Predict the aesthetic score for a given image.