Spaces:
				
			
			
	
			
			
		Running
		
			on 
			
			Zero
	
	
	
			
			
	
	
	
	
		
		
		Running
		
			on 
			
			Zero
	Clean debug strings.
Browse files
    	
        VideoLLaMA2/videollama2/model/videollama2_arch.py
    CHANGED
    
    | @@ -117,7 +117,6 @@ class Videollama2MetaForCausalLM(ABC): | |
| 117 |  | 
| 118 | 
             
                    data_batch = []
         | 
| 119 | 
             
                    for i, (data, modal) in enumerate(images):
         | 
| 120 | 
            -
                        print(data.shape, modal)
         | 
| 121 | 
             
                        if modal == 'image':
         | 
| 122 | 
             
                            data = data.expand(num_frames, -1, -1, -1)
         | 
| 123 | 
             
                        else:
         | 
| @@ -126,8 +125,6 @@ class Videollama2MetaForCausalLM(ABC): | |
| 126 |  | 
| 127 | 
             
                    data_batch = torch.stack(data_batch, dim=0)
         | 
| 128 |  | 
| 129 | 
            -
                    print(data_batch.shape)
         | 
| 130 | 
            -
             | 
| 131 | 
             
                    assert len(data_batch.size()) == 5
         | 
| 132 | 
             
                    batch_size = data_batch.size(0)
         | 
| 133 |  | 
|  | |
| 117 |  | 
| 118 | 
             
                    data_batch = []
         | 
| 119 | 
             
                    for i, (data, modal) in enumerate(images):
         | 
|  | |
| 120 | 
             
                        if modal == 'image':
         | 
| 121 | 
             
                            data = data.expand(num_frames, -1, -1, -1)
         | 
| 122 | 
             
                        else:
         | 
|  | |
| 125 |  | 
| 126 | 
             
                    data_batch = torch.stack(data_batch, dim=0)
         | 
| 127 |  | 
|  | |
|  | |
| 128 | 
             
                    assert len(data_batch.size()) == 5
         | 
| 129 | 
             
                    batch_size = data_batch.size(0)
         | 
| 130 |  |