syntax revision

augcog · Aug 13, 2024 · d1a3f0f · d1a3f0f
1 parent c636bab
commit d1a3f0f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ai_course_bot/ai-chatbot-backend/app/core/actions/llama_seletor.py b/ai_course_bot/ai-chatbot-backend/app/core/actions/llama_seletor.py
@@ -36,7 +36,7 @@ class Message(BaseModel):
     "text-generation",
     model=model_id,
     model_kwargs={"torch_dtype": torch.bfloat16},
-    device="cuda", 
+    device="cuda",
 )
 
 lock = threading.Lock()