|
My Project
|
Functions | |
| generate () | |
Variables | |
| app = Flask(__name__) | |
| int | PLUGIN_PARAM_COUNT = 9 |
| str | MODEL_PATH = "../saved_models/my_plugin_ai.pth" |
| list | PARAM_KEYS |
| device = torch.device('cpu') | |
| model = TextToParams(num_plugin_parameters=PLUGIN_PARAM_COUNT) | |
| tokenizer = AutoTokenizer.from_pretrained("prajjwal1/bert-tiny") | |
| port | |
| server.generate | ( | ) |
| server.app = Flask(__name__) |
| server.device = torch.device('cpu') |
| server.model = TextToParams(num_plugin_parameters=PLUGIN_PARAM_COUNT) |
| str server.MODEL_PATH = "../saved_models/my_plugin_ai.pth" |
| list server.PARAM_KEYS |
| int server.PLUGIN_PARAM_COUNT = 9 |
| server.port |
| server.tokenizer = AutoTokenizer.from_pretrained("prajjwal1/bert-tiny") |