.

main.py

@@ -38,8 +38,8 @@ LATENT_DIMS = (16, SCREEN_HEIGHT // 8, SCREEN_WIDTH // 8)
 # Initialize the model at the start of your application
 #model = initialize_model("config_csllm.yaml", "yuntian-deng/computer-model")
 #model = initialize_model("config_rnn.yaml", "yuntian-deng/computer-model")
-model = initialize_model("config_final_model.yaml", "yuntian-deng/computer-model-noss")
-
+#model = initialize_model("config_final_model.yaml", "yuntian-deng/computer-model-noss")
+model = initialize_model("config_final_model.yaml", "yuntian-deng/computer-model")
 model = model.to(device)
 #model = torch.compile(model)
 padding_image = torch.zeros(*LATENT_DIMS).unsqueeze(0).to(device)