Update llama2-model.libsonnet

GoogleCloudPlatform · Aug 16, 2023 · a726647 · a726647
1 parent 8e0ac8d
commit a726647
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/pytorch/nightly/llama2-model.libsonnet b/tests/pytorch/nightly/llama2-model.libsonnet
@@ -174,7 +174,7 @@ local utils = import 'templates/utils.libsonnet';
     },
   },
   local xla = self.xla,
-  stable:: common.PyTorchTpuVmMixin {
+  xla:: common.PyTorchTpuVmMixin {
     modelName+: '-xla',
     tpuSettings+: {
       tpuVmExtraSetup: |||