diff --git a/llama_stack/providers/impls/meta_reference/datasetio/datasetio.py b/llama_stack/providers/impls/meta_reference/datasetio/datasetio.py
index 7297f5d88..d0521a92d 100644
--- a/llama_stack/providers/impls/meta_reference/datasetio/datasetio.py
+++ b/llama_stack/providers/impls/meta_reference/datasetio/datasetio.py
@@ -69,8 +69,7 @@ class PandasDataframeDataset(BaseDataset):
         # check all columns in dataset schema are present
         assert len(self.df.columns) == len(self.dataset_def.dataset_schema)
 
-        # check all types match
-        print(self.df.dtypes)
+        # TODO: type checking against column types in dataset schema
 
     def load(self) -> None:
         if self.df is not None:
diff --git a/llama_stack/providers/registry/inference.py b/llama_stack/providers/registry/inference.py
index 065b5ce76..6f8bc2c6e 100644
--- a/llama_stack/providers/registry/inference.py
+++ b/llama_stack/providers/registry/inference.py
@@ -36,7 +36,7 @@ def available_providers() -> List[ProviderSpec]:
             pip_packages=(
                 META_REFERENCE_DEPS
                 + [
-                    "fbgemm-gpu==1.0.0",
+                    "fbgemm-gpu==0.8.0",
                 ]
             ),
             module="llama_stack.providers.impls.meta_reference.inference",