Spaces:
				
			
			
	
			
			
		Sleeping
		
	
	
	
			
			
	
	
	
	
		
		
		Sleeping
		
	Update app.py
Browse files
    	
        app.py
    CHANGED
    
    | @@ -41,7 +41,7 @@ bnb_config = transformers.BitsAndBytesConfig( | |
| 41 | 
             
            )
         | 
| 42 |  | 
| 43 | 
             
            tokenizer = AutoTokenizer.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct",token=HF_TOKEN)
         | 
| 44 | 
            -
            model = AutoModelForCausalLM.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct", device_map="auto",token=HF_TOKEN)  # to("cuda:0") 
         | 
| 45 | 
             
            terminators = [
         | 
| 46 | 
             
                tokenizer.eos_token_id,
         | 
| 47 | 
             
                tokenizer.convert_tokens_to_ids("<|eot_id|>")
         | 
|  | |
| 41 | 
             
            )
         | 
| 42 |  | 
| 43 | 
             
            tokenizer = AutoTokenizer.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct",token=HF_TOKEN)
         | 
| 44 | 
            +
            model = AutoModelForCausalLM.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct", device_map="auto",token=HF_TOKEN,quantization_config=bnb_config)  # to("cuda:0") 
         | 
| 45 | 
             
            terminators = [
         | 
| 46 | 
             
                tokenizer.eos_token_id,
         | 
| 47 | 
             
                tokenizer.convert_tokens_to_ids("<|eot_id|>")
         |