debug

app.py

@@ -14,7 +14,7 @@ MOCK = False
 TEST_FOLDER = "c4f5"
 
 MODEL_NAME="xu3kev/deepseekcoder-7b-logo-pbe"
-MODEL_NAME="meta-llama/Llama-2-7b" 
+MODEL_NAME="openlm-research/open_llama_3b" 
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 hug_model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, torch_dtype=torch.float16,).to('cuda')