This commit is contained in:
ebolam
2023-05-27 08:49:21 -04:00
parent cce5c1932c
commit 47276c3424

View File

@@ -339,7 +339,8 @@ class HFInferenceModel(InferenceModel):
Returns a string of the model's path locally, or None if it is not downloaded.
If ignore_existance is true, it will always return a path.
"""
if os.path.exists(self.path):
if self.path is not None:
if os.path.exists(self.path):
return self.path
if self.model_name in ["NeoCustom", "GPT2Custom", "TPUMeshTransformerGPTJ", "TPUMeshTransformerGPTNeoX"]: