diff --git a/CMakeLists.txt b/CMakeLists.txt
index bfcd083..698edbb 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -19,7 +19,6 @@ set(LIBRARY_SOURCES
     ${LIBRARY_SOURCES}
 )
 
-set(CMAKE_CUDA_ARCHITECTURES 75)
 set(CMAKE_EXPORT_COMPILE_COMMANDS ON)
 
 # Build static library
diff --git a/README.md b/README.md
index abc7c56..a3d178e 100644
--- a/README.md
+++ b/README.md
@@ -29,7 +29,7 @@ Convolutional Neural Network inference library running on CUDA.
 ```sh
 mkdir build
 cd build
-cmake -S ..
+cmake -S .. -DCMAKE_CUDA_ARCHITECTURES=75  # Replace with you cuda architecture
 make
 ```
 
diff --git a/tools/export_model_weights.py b/tools/export_model_weights.py
index e9f839e..4eb776a 100644
--- a/tools/export_model_weights.py
+++ b/tools/export_model_weights.py
@@ -25,4 +25,4 @@ def export_model_weights(model: torch.nn.Module, filename):
 
         f.seek(0)
         f.write(struct.pack('q', len(header)))            
-        f.write(header.encode('utf-8'))
\ No newline at end of file
+        f.write(header.encode('utf-8'))