commit changes

app.py

@@ -13,7 +13,7 @@ model_name = "TRI-ML/DCLM-1B-IT"
 # Load the configuration, tokenizer, and model separately  
 config = OpenLMConfig.from_pretrained(model_name)  
 tokenizer = AutoTokenizer.from_pretrained(model_name)  
-model = OpenLMforCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloat16, device_map="cuda", config=config)  
+model = OpenLMforCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloat16, device_map="cuda", config=config  )
   
 # Define the prompt format  
 def create_prompt(instruction):