Spaces:

rewgwrth
/

Game-Engine-3-LLM

Sleeping

rewgwrth commited on Oct 27, 2025

Commit

c31212e

1 Parent(s): 568586b

ge3

Files changed (1) hide show

main.py CHANGED Viewed

@@ -21,7 +21,7 @@ try:
     tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
     model = AutoModelForCausalLM.from_pretrained(
         MODEL_NAME,
-        torch_dtype=torch.float16,
         device_map="auto",
         low_cpu_mem_usage=True
     )

     tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
     model = AutoModelForCausalLM.from_pretrained(
         MODEL_NAME,
+        torch_dtype=torch.float32,
         device_map="auto",
         low_cpu_mem_usage=True
     )