diff --git a/skynet/dgpu/compute.py b/skynet/dgpu/compute.py
index 6b2f092..f069e60 100755
--- a/skynet/dgpu/compute.py
+++ b/skynet/dgpu/compute.py
@@ -7,17 +7,10 @@ import gc
 import logging
 
 from hashlib import sha256
-# import zipfile
-# from PIL import Image
-# from diffusers import DiffusionPipeline
 
 import trio
 import torch
 
-# from skynet.constants import (
-#     DEFAULT_INITAL_MODEL,
-#     MODELS,
-# )
 from skynet.dgpu.errors import (
     DGPUComputeError,
     DGPUInferenceCancelled,
diff --git a/skynet/dgpu/daemon.py b/skynet/dgpu/daemon.py
index c95e1ca..0f49ff7 100755
--- a/skynet/dgpu/daemon.py
+++ b/skynet/dgpu/daemon.py
@@ -1,11 +1,10 @@
-from datetime import datetime
-from functools import partial
-from hashlib import sha256
 import json
 import logging
 import random
-# import traceback
 import time
+from datetime import datetime
+from functools import partial
+from hashlib import sha256
 
 import trio
 from quart import jsonify
diff --git a/skynet/dgpu/network.py b/skynet/dgpu/network.py
index 654cec8..21ddfd8 100755
--- a/skynet/dgpu/network.py
+++ b/skynet/dgpu/network.py
@@ -2,7 +2,6 @@ import io
 import json
 import time
 import logging
-
 from pathlib import Path
 from functools import partial
 
@@ -11,12 +10,7 @@ import leap
 import anyio
 import httpx
 import outcome
-
-from PIL import (
-    Image,
-    # UnidentifiedImageError,  # TODO, remove?
-)
-
+from PIL import Image
 from leap.cleos import CLEOS
 from leap.protocol import Asset
 from skynet.constants import (
@@ -28,8 +22,6 @@ from skynet.ipfs import (
     AsyncIPFSHTTP,
     get_ipfs_file,
 )
-# TODO, remove?
-# from skynet.dgpu.errors import DGPUComputeError
 
 
 REQUEST_UPDATE_TIME: int = 3