Waw

my_text_to_text_dataset.py

@@ -7,7 +7,11 @@ def generate_text_to_text_data():
     # Your data generation logic here
     prompts = ["Tell a story about", "Describe a scene with", "Explain the concept of"]
 
-    your_model = YourTextGenerationModel()
+    vocab_size = 10000  # Replace with your actual vocabulary size
+    embedding_dim = 128  # Replace with your desired embedding dimension
+    hidden_dim = 256  # Replace with your desired hidden dimension
+
+    your_model = YourTextGenerationModel(vocab_size, embedding_dim, hidden_dim)
     generated_texts = [your_model.generate_text(prompt) for prompt in prompts]
 
     data = {