Spaces:

TULLUS
/

CodeParrot_MINI_LAB

Running on Zero

TULLUS commited on 2 days ago

Commit

19c39cc

verified ·

1 Parent(s): 7bddc65

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -26,7 +26,7 @@ from transformers import (
 # at module import time; only calls to generate() are inside @spaces.GPU.
 # ---------------------------------------------------------------------------
-MODEL_ID = "TULLUS/codeparrot-small-multi"
 CHECKPOINT_DTYPE = torch.float32  # Preserve the repository config.json dtype: F32.
 MAX_NEW_TOKENS_UI = 384
 MAX_CONTEXT_PROMPT_CHARS = 40_000  # Secondary guard before tokenization.

 # at module import time; only calls to generate() are inside @spaces.GPU.
 # ---------------------------------------------------------------------------
+MODEL_ID = "TULLUS/codeparrot-small-multi-F32"
 CHECKPOINT_DTYPE = torch.float32  # Preserve the repository config.json dtype: F32.
 MAX_NEW_TOKENS_UI = 384
 MAX_CONTEXT_PROMPT_CHARS = 40_000  # Secondary guard before tokenization.