Dean
/
MegaMolBART
mirror of https://github.com/NVIDIA/MegaMolBART


  
1

	
2

	
3

	
4

	
5

	
6

	
7

	
8

	
9

	
10

	
11

	
12

	
13

	
14

	
15

	
16

	
17

	
18

	
19

	
20

	
21

	
22

	
23

	
24

	
25

	
26

	
27

	
28

	
29

	
30

	
31

	
32

	
33

	
34

	
35

	
36

	
37

	
38

	
39

	
40

	
41

	
42

	
43

	
44

	
45

	
46

	
47

	
            # -*- coding: utf-8 -*-
# Generated by the protocol buffer compiler.  DO NOT EDIT!
# source: megamolbart.proto
"""Generated protocol buffer code."""
from google.protobuf import descriptor as _descriptor
from google.protobuf import descriptor_pool as _descriptor_pool
from google.protobuf import message as _message
from google.protobuf import reflection as _reflection
from google.protobuf import symbol_database as _symbol_database
# @@protoc_insertion_point(imports)

_sym_db = _symbol_database.Default()


DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x11megamolbart.proto\x12!nemo_chem.models.megamolbart.grpc\"L\n\tInputSpec\x12\x0c\n\x04smis\x18\x01 \x03(\t\x12\x15\n\rhidden_states\x18\x02 \x03(\x02\x12\x0b\n\x03\x64im\x18\x03 \x03(\x05\x12\r\n\x05masks\x18\x04 \x03(\x08\"a\n\nOutputSpec\x12\x0c\n\x04smis\x18\x01 \x03(\t\x12\x15\n\rhidden_states\x18\x02 \x03(\x02\x12\x12\n\nembeddings\x18\x03 \x03(\x02\x12\x0b\n\x03\x64im\x18\x04 \x03(\x05\x12\r\n\x05masks\x18\x05 \x03(\x08\x32\xe7\x02\n\x11GenerativeSampler\x12r\n\x11SmilesToEmbedding\x12,.nemo_chem.models.megamolbart.grpc.InputSpec\x1a-.nemo_chem.models.megamolbart.grpc.OutputSpec\"\x00\x12o\n\x0eSmilesToHidden\x12,.nemo_chem.models.megamolbart.grpc.InputSpec\x1a-.nemo_chem.models.megamolbart.grpc.OutputSpec\"\x00\x12m\n\x0cHiddenToSmis\x12,.nemo_chem.models.megamolbart.grpc.InputSpec\x1a-.nemo_chem.models.megamolbart.grpc.OutputSpec\"\x00\x62\x06proto3')


_INPUTSPEC = DESCRIPTOR.message_types_by_name['InputSpec']
_OUTPUTSPEC = DESCRIPTOR.message_types_by_name['OutputSpec']
InputSpec = _reflection.GeneratedProtocolMessageType('InputSpec', (_message.Message,), {
  'DESCRIPTOR' : _INPUTSPEC,
  '__module__' : 'megamolbart_pb2'
  # @@protoc_insertion_point(class_scope:nemo_chem.models.megamolbart.grpc.InputSpec)
  })
_sym_db.RegisterMessage(InputSpec)

OutputSpec = _reflection.GeneratedProtocolMessageType('OutputSpec', (_message.Message,), {
  'DESCRIPTOR' : _OUTPUTSPEC,
  '__module__' : 'megamolbart_pb2'
  # @@protoc_insertion_point(class_scope:nemo_chem.models.megamolbart.grpc.OutputSpec)
  })
_sym_db.RegisterMessage(OutputSpec)

_GENERATIVESAMPLER = DESCRIPTOR.services_by_name['GenerativeSampler']
if _descriptor._USE_C_DESCRIPTORS == False:

  DESCRIPTOR._options = None
  _INPUTSPEC._serialized_start=56
  _INPUTSPEC._serialized_end=132
  _OUTPUTSPEC._serialized_start=134
  _OUTPUTSPEC._serialized_end=231
  _GENERATIVESAMPLER._serialized_start=234
  _GENERATIVESAMPLER._serialized_end=593
# @@protoc_insertion_point(module_scope)