fix: deploy key gen

higgsfield-ai · Feb 11, 2024 · f7a4fbf · f7a4fbf
1 parent 0cde64b
commit f7a4fbf
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/higgsfield/internal/init.py b/higgsfield/internal/init.py
@@ -76,13 +76,13 @@ def generate_deploy_keys() -> Tuple[bytes, bytes]:
 
     private_bytes = private_key.private_bytes(
         encoding=serialization.Encoding.PEM,
-        format=serialization.PrivateFormat.PKCS8,
+        format=serialization.PrivateFormat.OpenSSH,
         encryption_algorithm=serialization.NoEncryption(),
     )
 
     public_bytes = public_key.public_bytes(
-        encoding=serialization.Encoding.PEM,
-        format=serialization.PublicFormat.SubjectPublicKeyInfo,
+        encoding=serialization.Encoding.OpenSSH,
+        format=serialization.PublicFormat.OpenSSH,
     )
 
     return private_bytes, public_bytes