From d9dda67c98d7feedfdbf628e55d3cca603b487cb Mon Sep 17 00:00:00 2001 From: "DESKTOP-CNFQ7PM\\Primere" Date: Mon, 28 Apr 2025 09:03:54 +0200 Subject: [PATCH] V 1.0.7 - LLM enhancer - model device --- components/llm_enhancer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/components/llm_enhancer.py b/components/llm_enhancer.py index 5b6aa4e..767ab23 100644 --- a/components/llm_enhancer.py +++ b/components/llm_enhancer.py @@ -328,7 +328,7 @@ class PromptEnhancerLLM: ) messages = create_direct_template(create_user_prompt(simple_caption)) - input_ids = self.tokenizer.encode(messages, return_tensors="pt").to(self.device) + input_ids = self.tokenizer.encode(messages, return_tensors="pt").to(self.model.device) # streamer = TextStreamer(self.tokenizer, skip_special_tokens=True, clean_up_tokenization_spaces=True) output = self.model.generate(