Update base for Update on "Update phi-3-mini to use the export library"

helunwencser · helunwencser · commit c0c0f5ea1b14 · 2024-07-10T17:41:39.000-07:00
as title. Note: exporting model from HF is currently blocked by this issue, https://fburl.com/qpaapajr. We need to apply the following trick to export the model. ``` model = torch.export._trace._export( model, example_inputs, dynamic_shapes=dynamic_shape, strict=False, pre_dispatch=False, ) ``` Differential Revision: [D59503255](https://our.internmc.facebook.com/intern/diff/D59503255/) [ghstack-poisoned]
diff --git a/examples/models/phi-3-mini/export_phi-3-mini.py b/examples/models/phi-3-mini/export_phi-3-mini.py
@@ -20,9 +20,8 @@
     XNNPACKQuantizer,
 )
 
-from transformers import (  # @manual=fbsource//third-party/pypi/transformers:transformers
-    Phi3ForCausalLM,
-)
+# pyre-ignore: Undefined attribute [16]: Module `transformers` has no attribute `Phi3ForCausalLM`
+from transformers import Phi3ForCausalLM
 
 
 def main() -> None:

Original file line number	Diff line number	Diff line change
`@@ -20,9 +20,8 @@`
`20`	`20`	`XNNPACKQuantizer,`
`21`	`21`	`)`
`22`	`22`
`23`		`-from transformers import ( # @manual=fbsource//third-party/pypi/transformers:transformers`
`24`		`- Phi3ForCausalLM,`
`25`		`-)`
	`23`	+# pyre-ignore: Undefined attribute [16]: Module `transformers` has no attribute `Phi3ForCausalLM`
	`24`	`+from transformers import Phi3ForCausalLM`
`26`	`25`
`27`	`26`
`28`	`27`	`def main() -> None:`