diff --git a/lollms/utilities.py b/lollms/utilities.py
index 3bbb106..8041bce 100644
--- a/lollms/utilities.py
+++ b/lollms/utilities.py
@@ -29,6 +29,7 @@ import yaml
 import asyncio
 
 import ctypes
+import io
 
 def is_asyncio_loop_running():
     try: