Catch OSError while trying to load cached model

models.py

@@ -535,7 +535,7 @@ class StableAudWrapper(PipelineWrapper):
         try:
             self.model = StableAudioPipeline.from_pretrained(self.model_id, token=self.token, local_files_only=True
                                                              ).to(self.device)
-        except FileNotFoundError:
+        except (FileNotFoundError, OSError):
             self.model = StableAudioPipeline.from_pretrained(self.model_id, token=self.token, local_files_only=False
                                                              ).to(self.device)
         self.model.transformer.eval()