File size: 356 Bytes
7b127f4
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
from app.ModelDirector import ModelDirector

class SketchDirector(ModelDirector):
    def get_ae_weights(self):
        return 'YuXingyao/HoLa-Brep/AE_deepcad_1100k.ckpt'
    
    def get_diffusion_weights(self):
        return 'YuXingyao/HoLa-Brep/Diffusion_sketch_sq30_1500k.ckpt'
    
    def get_generating_condition(self):
        return 'sketch'