Update app.py

app.py

@@ -18,10 +18,10 @@ model = AutoPeftModelForCausalLM.from_pretrained(
     low_cpu_mem_usage=True,
     return_dict=True,
     torch_dtype=torch.float16,
-    device_map="auto",
+    device_map="cpu",
     trust_remote_code=True
     )
-model.tie_weights()
+
 tokenizer = AutoTokenizer.from_pretrained("Aneeth/zephyr_10k")
 generation_config = GenerationConfig(
     do_sample=True,