Create Dockerfile

Dockerfile

@@ -0,0 +1,48 @@
+import docker
+from transformers import DollyTokenizer, DollyForSequenceClassification
+
+# Initialize the Docker client
+client = docker.from_env()
+
+# Create a new container based on the Dolly LLM image
+container = client.containers.create(
+    image="huggingface/dolly-base-uncased",
+    detach=True,
+    command=[
+        "python",
+        "-m",
+        "transformers.dolly.run_dolly",
+        "--model_name",
+        "dolly-base-uncased",
+        "--task",
+        "sequence_classification",
+        "--tokenizer",
+        "dolly-base-uncased",
+        "--max_seq_len",
+        512,
+        "--per_device_train_batch_size",
+        16,
+        "--per_device_eval_batch_size",
+        64,
+        "--evaluation_strategy",
+        "epoch",
+        "--learning_rate",
+        5e-5,
+        "--num_train_epochs",
+        3,
+        "--save_total_limit",
+        2,
+        "--save_steps",
+        500,
+        "--load_best_model_at_end",
+        "true",
+        "--tensorboard_log_dir",
+        "/logs",
+    ]
+)
+
+# Start the container
+container.start()
+
+# Wait for the container to finish
+container.wait()