Spaces:
				
			
			
	
			
			
		Runtime error
		
	
	
	
			
			
	
	
	
	
		
		
		Runtime error
		
	add lmsys vicuna to the arena
Browse files
    	
        app.py
    CHANGED
    
    | @@ -93,8 +93,6 @@ class Pipeline: | |
| 93 | 
             
                                return
         | 
| 94 | 
             
                        elif response.status_code >= 400:
         | 
| 95 | 
             
                            logging.error(response.json())
         | 
| 96 | 
            -
                        # Sleep for 0.5 seconds between each request
         | 
| 97 | 
            -
                        sleep(0.5)
         | 
| 98 |  | 
| 99 | 
             
                def poll_for_status(self, task_id):
         | 
| 100 | 
             
                    url = f"https://api.runpod.ai/v2/{self.endpoint_id}/status/{task_id}"
         | 
| @@ -122,6 +120,7 @@ AVAILABLE_MODELS = { | |
| 122 | 
             
                "manticore-13b-chat": ("u6tv84bpomhfei", prompt_chat),
         | 
| 123 | 
             
                "airoboros-13b": ("rglzxnk80660ja", prompt_chat),
         | 
| 124 | 
             
                "wizard-vicuna-13b": ("9vvpikt4ttyqos", prompt_chat),
         | 
|  | |
| 125 | 
             
                "supercot-13b": ("0be7865dwxpwqk", prompt_instruct, ["Instruction:"]),
         | 
| 126 | 
             
                "mpt-7b-instruct": ("jpqbvnyluj18b0", prompt_instruct),
         | 
| 127 | 
             
            }
         | 
|  | |
| 93 | 
             
                                return
         | 
| 94 | 
             
                        elif response.status_code >= 400:
         | 
| 95 | 
             
                            logging.error(response.json())
         | 
|  | |
|  | |
| 96 |  | 
| 97 | 
             
                def poll_for_status(self, task_id):
         | 
| 98 | 
             
                    url = f"https://api.runpod.ai/v2/{self.endpoint_id}/status/{task_id}"
         | 
|  | |
| 120 | 
             
                "manticore-13b-chat": ("u6tv84bpomhfei", prompt_chat),
         | 
| 121 | 
             
                "airoboros-13b": ("rglzxnk80660ja", prompt_chat),
         | 
| 122 | 
             
                "wizard-vicuna-13b": ("9vvpikt4ttyqos", prompt_chat),
         | 
| 123 | 
            +
                "lmsys-vicuna-13b": ("2nlb32ydkaz6yd", prompt_chat),
         | 
| 124 | 
             
                "supercot-13b": ("0be7865dwxpwqk", prompt_instruct, ["Instruction:"]),
         | 
| 125 | 
             
                "mpt-7b-instruct": ("jpqbvnyluj18b0", prompt_instruct),
         | 
| 126 | 
             
            }
         | 
