Spaces:
				
			
			
	
			
			
		Runtime error
		
	
	
	
			
			
	
	
	
	
		
		
		Runtime error
		
	Update app.py
Browse files
    	
        app.py
    CHANGED
    
    | @@ -15,6 +15,7 @@ DESCRIPTION = "# ICONN Lite Chat" | |
| 15 | 
             
            if not torch.cuda.is_available():
         | 
| 16 | 
             
                DESCRIPTION += "\n<p class='warning'>Running on CPU 🥶 This demo does not work on CPU.</p>"
         | 
| 17 |  | 
|  | |
| 18 | 
             
            MAX_MAX_NEW_TOKENS = 100000000
         | 
| 19 | 
             
            DEFAULT_MAX_NEW_TOKENS = 10240
         | 
| 20 | 
             
            MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "4096"))
         | 
|  | |
| 15 | 
             
            if not torch.cuda.is_available():
         | 
| 16 | 
             
                DESCRIPTION += "\n<p class='warning'>Running on CPU 🥶 This demo does not work on CPU.</p>"
         | 
| 17 |  | 
| 18 | 
            +
            top_k: int = 50
         | 
| 19 | 
             
            MAX_MAX_NEW_TOKENS = 100000000
         | 
| 20 | 
             
            DEFAULT_MAX_NEW_TOKENS = 10240
         | 
| 21 | 
             
            MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "4096"))
         | 
 
			
