Spaces:
				
			
			
	
			
			
		Runtime error
		
	
	
	
			
			
	
	
	
	
		
		
		Runtime error
		
	Commit 
							
							·
						
						db20d3e
	
1
								Parent(s):
							
							23e4ec1
								
limit chunk size
Browse files- app.py +1 -0
- tsr/system.py +1 -1
    	
        app.py
    CHANGED
    
    | @@ -25,6 +25,7 @@ model = TSR.from_pretrained( | |
| 25 | 
             
                weight_name="model.ckpt",
         | 
| 26 | 
             
                token=HF_TOKEN
         | 
| 27 | 
             
            )
         | 
|  | |
| 28 | 
             
            model.to(device)
         | 
| 29 |  | 
| 30 | 
             
            rembg_session = rembg.new_session()
         | 
|  | |
| 25 | 
             
                weight_name="model.ckpt",
         | 
| 26 | 
             
                token=HF_TOKEN
         | 
| 27 | 
             
            )
         | 
| 28 | 
            +
            model.renderer.set_chunk_size(131072)
         | 
| 29 | 
             
            model.to(device)
         | 
| 30 |  | 
| 31 | 
             
            rembg_session = rembg.new_session()
         | 
    	
        tsr/system.py
    CHANGED
    
    | @@ -168,7 +168,7 @@ class TSR(BaseModule): | |
| 168 | 
             
                        return
         | 
| 169 | 
             
                    self.isosurface_helper = MarchingCubeHelper(resolution)
         | 
| 170 |  | 
| 171 | 
            -
                def extract_mesh(self, scene_codes, resolution: int = 256, threshold: float =  | 
| 172 | 
             
                    self.set_marching_cubes_resolution(resolution)
         | 
| 173 | 
             
                    meshes = []
         | 
| 174 | 
             
                    for scene_code in scene_codes:
         | 
|  | |
| 168 | 
             
                        return
         | 
| 169 | 
             
                    self.isosurface_helper = MarchingCubeHelper(resolution)
         | 
| 170 |  | 
| 171 | 
            +
                def extract_mesh(self, scene_codes, resolution: int = 256, threshold: float = 25.0):
         | 
| 172 | 
             
                    self.set_marching_cubes_resolution(resolution)
         | 
| 173 | 
             
                    meshes = []
         | 
| 174 | 
             
                    for scene_code in scene_codes:
         | 
