small fixes

Author: wizeng23

notebooks/Oumi - Using vLLM Engine for Inference.ipynb

@@ -88,9 +88,9 @@
     "%%writefile llama70b_inference_config.yaml\n",
     "\n",
     "model:\n",
-    "#   model_name: \"meta-llama/Meta-Llama-3.1-8B-Instruct\"  # 8B model, requires 1x A100-40GB GPUs\n",
-    "  model_name: \"meta-llama/Meta-Llama-3.1-70B-Instruct\"  # 70B model, requires 4x A100-40GB GPUs\n",
-    "#   model_name: \"bartowski/Meta-Llama-3.1-70B-Instruct-GGUF\"  # 4-bit quantized model, requires 1x A100-40GB GPUs. See bonus section for more details.\n",
+    "  model_name: \"meta-llama/Meta-Llama-3.1-8B-Instruct\"  # 8B model, requires 1x A100-40GB GPUs\n",
+    "  # model_name: \"meta-llama/Meta-Llama-3.1-70B-Instruct\"  # 70B model, requires 4x A100-40GB GPUs\n",
+    "  # model_name: \"bartowski/Meta-Llama-3.1-70B-Instruct-GGUF\"  # 4-bit quantized model, requires 1x A100-40GB GPUs. See bonus section for more details.\n",
     "  model_max_length: 512\n",
     "  torch_dtype_str: \"bfloat16\"\n",
     "  trust_remote_code: True\n",

src/oumi/core/types/turn.py

@@ -107,8 +107,8 @@ def is_text(self) -> bool:
         return self.type == Type.TEXT
 
     def __repr__(self):
-        """Return a string representation of the message."""
-        content = self.content if self.is_text() else "BINARY"
+        """Returns a string representation of the message."""
+        content = self.content if self.is_text() else "<non-text-content>"
         return f"{self.role.upper()}: {content}"
 
 
@@ -133,7 +133,7 @@ class Conversation(pydantic.BaseModel):
     """
 
     def __getitem__(self, idx: int) -> Message:
-        """Get the message at the specified index.
+        """Gets the message at the specified index.
 
         Args:
             idx (int): The index of the message to retrieve.
@@ -144,7 +144,7 @@ def __getitem__(self, idx: int) -> Message:
         return self.messages[idx]
 
     def first_message(self, role: Optional[Role] = None) -> Optional[Message]:
-        """Get the first message in the conversation, optionally filtered by role.
+        """Gets the first message in the conversation, optionally filtered by role.
 
         Args:
             role: The role to filter messages by.
@@ -158,7 +158,7 @@ def first_message(self, role: Optional[Role] = None) -> Optional[Message]:
         return messages[0] if len(messages) > 0 else None
 
     def last_message(self, role: Optional[Role] = None) -> Optional[Message]:
-        """Get the last message in the conversation, optionally filtered by role.
+        """Gets the last message in the conversation, optionally filtered by role.
 
         Args:
             role: The role to filter messages by.
@@ -172,7 +172,7 @@ def last_message(self, role: Optional[Role] = None) -> Optional[Message]:
         return messages[-1] if len(messages) > 0 else None
 
     def filter_messages(self, role: Optional[Role] = None) -> List[Message]:
-        """Get all messages in the conversation, optionally filtered by role.
+        """Gets all messages in the conversation, optionally filtered by role.
 
         Args:
             role: The role to filter messages by.
@@ -188,7 +188,7 @@ def filter_messages(self, role: Optional[Role] = None) -> List[Message]:
         return messages
 
     def __repr__(self):
-        """Return a string representation of the conversation."""
+        """Returns a string representation of the conversation."""
         return "\n".join([repr(m) for m in self.messages])