Spaces:

ntaexams
/

ProfCool

Sleeping

ntaexams commited on Mar 9, 2025

Commit

f0b2b62

verified ·

1 Parent(s): dd79d33

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ model = AutoModelForCausalLM.from_pretrained(
     model_file=model_file,
     model_type="mistral",
     # Uncomment if GPU is available
-    gpu_layers=50
     context_length=1024
 )
 print("Model downloaded and loaded successfully.")

     model_file=model_file,
     model_type="mistral",
     # Uncomment if GPU is available
+    # gpu_layers=50
     context_length=1024
 )
 print("Model downloaded and loaded successfully.")