fix: accept paths as AsRef<Path>

pykeio · May 8, 2023 · 0c94307 · 0c94307
1 parent 34c6d84
commit 0c94307
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/pipelines/stable_diffusion/text_embeddings.rs b/src/pipelines/stable_diffusion/text_embeddings.rs
@@ -49,7 +49,7 @@ impl TextEmbeddings {
 		Ok(Self { tokenizer, text_hidden_size, tokens })
 	}
 
-	pub fn add_token_from_file<P: AsRef<PathBuf>>(&mut self, path: P) -> io::Result<AddedToken> {
+	pub fn add_token_from_file<P: AsRef<Path>>(&mut self, path: P) -> io::Result<AddedToken> {
 		let path = path.as_ref();
 		self.add_token_from_reader(BufReader::new(File::open(path)?))
 	}