surrogate1 commited on
Commit
dc8decc
·
verified ·
1 Parent(s): 3434697

fix: revert to 7B INT4 (14B caused RUNTIME_ERROR on A10G)

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ import gradio as gr
15
  import spaces
16
  import torch
17
 
18
- BASE_MODEL = os.environ.get("BASE_MODEL", "Qwen/Qwen2.5-Coder-14B-Instruct")
19
  HF_TOKEN = os.environ.get("HF_TOKEN", "")
20
 
21
  DOMAIN_SEEDS = {
 
15
  import spaces
16
  import torch
17
 
18
+ BASE_MODEL = os.environ.get("BASE_MODEL", "Qwen/Qwen2.5-Coder-7B-Instruct")
19
  HF_TOKEN = os.environ.get("HF_TOKEN", "")
20
 
21
  DOMAIN_SEEDS = {