ByteEd
/

3DModelGeneration_PointCloud

Model card Files Files and versions Community

Thomas Male commited on Feb 28, 2024

Commit

3893e21

·

verified ·

1 Parent(s): eb11cf4

Update handler.py

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -24,8 +24,8 @@ class EndpointHandler():
         # load the optimized model
         print('creating base model...')
-        #self.base_name = 'base40M-textvec'
-        self.base_name = 'base40M'
         self.base_model = model_from_config(MODEL_CONFIGS[self.base_name], device)
         self.base_model.eval()

         # load the optimized model
         print('creating base model...')
+        self.base_name = 'base40M-textvec'
+        #self.base_name = 'base40M'
         self.base_model = model_from_config(MODEL_CONFIGS[self.base_name], device)
         self.base_model.eval()