Update main.py
Browse files
    	
        main.py
    CHANGED
    
    | 
         @@ -50,7 +50,7 @@ def LLModel(): 
     | 
|
| 50 | 
         
             
                os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.environ['HUGGINGFACEHUB_API_TOKEN']
         
     | 
| 51 | 
         
             
                repo_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
         
     | 
| 52 | 
         
             
                llm = HuggingFaceEndpoint(
         
     | 
| 53 | 
         
            -
                    repo_id=repo_id, max_new_tokens= 
     | 
| 54 | 
         
             
                )
         
     | 
| 55 | 
         
             
                add_llm_provider(
         
     | 
| 56 | 
         
             
                    LangchainGenericProvider(
         
     | 
| 
         | 
|
| 50 | 
         
             
                os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.environ['HUGGINGFACEHUB_API_TOKEN']
         
     | 
| 51 | 
         
             
                repo_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
         
     | 
| 52 | 
         
             
                llm = HuggingFaceEndpoint(
         
     | 
| 53 | 
         
            +
                    repo_id=repo_id, max_new_tokens=5300, temperature=1.0, task="text2text-generation", streaming=True
         
     | 
| 54 | 
         
             
                )
         
     | 
| 55 | 
         
             
                add_llm_provider(
         
     | 
| 56 | 
         
             
                    LangchainGenericProvider(
         
     |