Upload AutoEncoder

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:214c2fd1c62f2246bbf01150156b1eae89b22a382f5e0f8d026e86f759f2367f
+oid sha256:a0bab99ecd0c72cac40bdba7d68fb1827267512ef500ace097555243f9f6cab7
 size 133840

modeling_autoencoder.py

@@ -6,22 +6,27 @@ from torch import nn, Tensor
 from transformers import PretrainedConfig, PreTrainedModel, AutoConfig, AutoModel
 from transformers.utils import ModelOutput
 
-# from huggingface_hub import notebook_login
+#from huggingface_hub import notebook_login
+#from transformers import AutoConfig, AutoModel
+#from autoencoder_model.modeling_autoencoder import AutoEncoder, AutoEncoderConfig
 
-# notebook_login()
+#notebook_login()
 
-# AutoEncoderConfig.register_for_auto_class()
-# AutoEncoder.register_for_auto_class("AutoModel")
+# Register Huggingface Model
+#AutoEncoderConfig.register_for_auto_class()
+#AutoEncoder.register_for_auto_class("AutoModel")
 
-# AutoConfig.register("autoencoder", AutoEncoderConfig)
-# AutoModel.register(AutoEncoderConfig, AutoModel)
+#AutoConfig.register("autoencoder", AutoEncoderConfig)
+#AutoModel.register(AutoEncoderConfig, AutoModel)
 
-# autoencoder.push_to_hub("autoencoder")
+# Create Model
+#autoencoder = AutoEncoder(AutoEncoderConfig())
+#autoencoder.push_to_hub("autoencoder")
+
+# Download Model
+#config = AutoConfig.from_pretrained("amaye15/autoencoder", trust_remote_code = True)
+#autoencoder = AutoModel.from_config(config, trust_remote_code = True)
 
-# from transformers import AutoConfig, AutoModel
- 
-# config = AutoConfig.from_pretrained("amaye15/autoencoder", trust_remote_code = True)
-# autoencoder = AutoModel.from_config(config, trust_remote_code = True)
 
 
 # Stucture