Spaces:
				
			
			
	
			
			
		Runtime error
		
	
	
	
			
			
	
	
	
	
		
		
		Runtime error
		
	Update model.py
Browse files
    	
        model.py
    CHANGED
    
    | @@ -85,10 +85,10 @@ class Model: | |
| 85 | 
             
                    ).to(device='cpu', dtype=precision)
         | 
| 86 |  | 
| 87 | 
             
                    model_weight = load_file(gpt_ckpt)
         | 
| 88 | 
            -
                     | 
| 89 | 
             
                    gpt_model.load_state_dict(model_weight, strict=True)
         | 
| 90 | 
             
                    gpt_model.eval()
         | 
| 91 | 
            -
                     | 
| 92 | 
             
                    print("gpt model is loaded")
         | 
| 93 | 
             
                    return gpt_model
         | 
| 94 |  | 
|  | |
| 85 | 
             
                    ).to(device='cpu', dtype=precision)
         | 
| 86 |  | 
| 87 | 
             
                    model_weight = load_file(gpt_ckpt)
         | 
| 88 | 
            +
                    print("prev:", model_weight['adapter.model.embeddings.patch_embeddings.projection.weight'])
         | 
| 89 | 
             
                    gpt_model.load_state_dict(model_weight, strict=True)
         | 
| 90 | 
             
                    gpt_model.eval()
         | 
| 91 | 
            +
                    print("loaded:", gpt_model.adapter.model.embeddings.patch_embeddings.projection.weight)
         | 
| 92 | 
             
                    print("gpt model is loaded")
         | 
| 93 | 
             
                    return gpt_model
         | 
| 94 |  | 
