@@ -18,33 +18,33 @@ pub enum MachineComprehension {
1818 ///
1919 /// > This model is a neural network for answering a query about a given context paragraph.
2020 ///
21- /// Source: [https://github.com/onnx/models/tree/master /text/machine_comprehension/bidirectional_attention_flow](https://github.com/onnx/models/tree/master /text/machine_comprehension/bidirectional_attention_flow)
21+ /// Source: [https://github.com/onnx/models/tree/main /text/machine_comprehension/bidirectional_attention_flow](https://github.com/onnx/models/tree/main /text/machine_comprehension/bidirectional_attention_flow)
2222 ///
2323 /// Variant downloaded: ONNX Version 1.4 with Opset Version 9.
2424 BiDAF ,
2525 /// Answers questions based on the context of the given input paragraph.
2626 ///
27- /// Source: [https://github.com/onnx/models/tree/master /text/machine_comprehension/bert-squad](https://github.com/onnx/models/tree/master /text/machine_comprehension/bert-squad)
27+ /// Source: [https://github.com/onnx/models/tree/main /text/machine_comprehension/bert-squad](https://github.com/onnx/models/tree/main /text/machine_comprehension/bert-squad)
2828 ///
2929 /// Variant downloaded: ONNX Version 1.5 with Opset Version 10.
3030 BERTSquad ,
3131 /// Large transformer-based model that predicts sentiment based on given input text.
3232 ///
3333 /// > Transformer-based language model for text generation.
3434 ///
35- /// Source: [https://github.com/onnx/models/tree/master /text/machine_comprehension/roberta](https://github.com/onnx/models/tree/master /text/machine_comprehension/roberta)
35+ /// Source: [https://github.com/onnx/models/tree/main /text/machine_comprehension/roberta](https://github.com/onnx/models/tree/main /text/machine_comprehension/roberta)
3636 RoBERTa ( RoBERTa ) ,
3737 /// Large transformer-based language model that given a sequence of words within some text, predicts the next word.
3838 ///
39- /// Source: [https://github.com/onnx/models/tree/master /text/machine_comprehension/gpt-2](https://github.com/onnx/models/tree/master /text/machine_comprehension/gpt-2)
39+ /// Source: [https://github.com/onnx/models/tree/main /text/machine_comprehension/gpt-2](https://github.com/onnx/models/tree/main /text/machine_comprehension/gpt-2)
4040 GPT2 ( GPT2 ) ,
4141}
4242
4343/// Large transformer-based model that predicts sentiment based on given input text.
4444///
4545/// > Transformer-based language model for text generation.
4646///
47- /// Source: [https://github.com/onnx/models/tree/master /text/machine_comprehension/roberta](https://github.com/onnx/models/tree/master /text/machine_comprehension/roberta)
47+ /// Source: [https://github.com/onnx/models/tree/main /text/machine_comprehension/roberta](https://github.com/onnx/models/tree/main /text/machine_comprehension/roberta)
4848#[ derive( Debug , Clone ) ]
4949pub enum RoBERTa {
5050 /// Variant with input is a sequence of words as a string. Example: "Text to encode: Hello, World"
@@ -61,7 +61,7 @@ pub enum RoBERTa {
6161///
6262/// > Transformer-based language model for text generation.
6363///
64- /// Source: [https://github.com/onnx/models/tree/master /text/machine_comprehension/gpt-2](https://github.com/onnx/models/tree/master /text/machine_comprehension/gpt-2)
64+ /// Source: [https://github.com/onnx/models/tree/main /text/machine_comprehension/gpt-2](https://github.com/onnx/models/tree/main /text/machine_comprehension/gpt-2)
6565///
6666/// Variant downloaded: ONNX Version 1.6 with Opset Version 10.
6767#[ derive( Debug , Clone ) ]
@@ -70,16 +70,16 @@ pub enum GPT2 {
7070 GPT2 ,
7171 /// GPT2 + script changes
7272 ///
73- /// See [https://github.com/onnx/models/blob/master /text/machine_comprehension/gpt-2/dependencies/GPT2-export.py](https://github.com/onnx/models/blob/master /text/machine_comprehension/gpt-2/dependencies/GPT2-export.py)
73+ /// See [https://github.com/onnx/models/blob/main /text/machine_comprehension/gpt-2/dependencies/GPT2-export.py](https://github.com/onnx/models/blob/main /text/machine_comprehension/gpt-2/dependencies/GPT2-export.py)
7474 /// for the script changes.
7575 GPT2LmHead ,
7676}
7777
7878impl ModelUrl for MachineComprehension {
7979 fn fetch_url ( & self ) -> & ' static str {
8080 match self {
81- MachineComprehension :: BiDAF => "https://github.com/onnx/models/raw/master /text/machine_comprehension/bidirectional_attention_flow/model/bidaf-9.onnx" ,
82- MachineComprehension :: BERTSquad => "https://github.com/onnx/models/raw/master /text/machine_comprehension/bert-squad/model/bertsquad-10.onnx" ,
81+ MachineComprehension :: BiDAF => "https://github.com/onnx/models/raw/main /text/machine_comprehension/bidirectional_attention_flow/model/bidaf-9.onnx" ,
82+ MachineComprehension :: BERTSquad => "https://github.com/onnx/models/raw/main /text/machine_comprehension/bert-squad/model/bertsquad-10.onnx" ,
8383 MachineComprehension :: RoBERTa ( variant) => variant. fetch_url ( ) ,
8484 MachineComprehension :: GPT2 ( variant) => variant. fetch_url ( ) ,
8585 }
@@ -89,17 +89,17 @@ impl ModelUrl for MachineComprehension {
8989impl ModelUrl for RoBERTa {
9090 fn fetch_url ( & self ) -> & ' static str {
9191 match self {
92- RoBERTa :: RoBERTaBase => "https://github.com/onnx/models/raw/master /text/machine_comprehension/roberta/model/roberta-base-11.onnx" ,
93- RoBERTa :: RoBERTaSequenceClassification => "https://github.com/onnx/models/raw/master /text/machine_comprehension/roberta/model/roberta-sequence-classification-9.onnx" ,
92+ RoBERTa :: RoBERTaBase => "https://github.com/onnx/models/raw/main /text/machine_comprehension/roberta/model/roberta-base-11.onnx" ,
93+ RoBERTa :: RoBERTaSequenceClassification => "https://github.com/onnx/models/raw/main /text/machine_comprehension/roberta/model/roberta-sequence-classification-9.onnx" ,
9494 }
9595 }
9696}
9797
9898impl ModelUrl for GPT2 {
9999 fn fetch_url ( & self ) -> & ' static str {
100100 match self {
101- GPT2 :: GPT2 => "https://github.com/onnx/models/raw/master /text/machine_comprehension/gpt-2/model/gpt2-10.onnx" ,
102- GPT2 :: GPT2LmHead => "https://github.com/onnx/models/raw/master /text/machine_comprehension/gpt-2/model/gpt2-lm-head-10.onnx" ,
101+ GPT2 :: GPT2 => "https://github.com/onnx/models/raw/main /text/machine_comprehension/gpt-2/model/gpt2-10.onnx" ,
102+ GPT2 :: GPT2LmHead => "https://github.com/onnx/models/raw/main /text/machine_comprehension/gpt-2/model/gpt2-lm-head-10.onnx" ,
103103 }
104104 }
105105}
0 commit comments