Update app.py

app.py

@@ -7,8 +7,8 @@ from transformers import TextStreamer
 import spaces
 
 # Load model and tokenizer
-model = AutoModelForCausalLM.from_pretrained("Rorical/0-roleplay", return_dict=True, trust_remote_code=True, low_cpu_mem_usage=False)
-tokenizer = AutoTokenizer.from_pretrained("Rorical/0-roleplay", trust_remote_code=True, low_cpu_mem_usage=False)
+model = AutoModelForCausalLM.from_pretrained("Rorical/0-roleplay", return_dict=True, trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained("Rorical/0-roleplay", trust_remote_code=True)
 tokenizer.chat_template = "{% for message in messages %}{{'' + ((message['role'] + '\n') if message['role'] != '' else '') + message['content'] + '' + '\n'}}{% endfor %}{% if add_generation_prompt %}{{ '星野\n' }}{% endif %}"
 
 # Define the response function