SohanAnanthula252 commited on
Commit
9f27054
·
verified ·
1 Parent(s): fcbf06a

Deploy Qwen2.5 Coder 3B Gradio app

Browse files
Files changed (3) hide show
  1. README.md +2 -2
  2. app.py +0 -1
  3. requirements.txt +0 -1
README.md CHANGED
@@ -14,5 +14,5 @@ pinned: false
14
  A coding assistant using Qwen2.5-Coder-3B-Instruct with Hugging Face ZeroGPU.
15
  The model is downloaded from Hugging Face on first startup. Inference requests
16
  use the largest ZeroGPU allocation; xlarge requests consume twice the daily
17
- GPU quota of the default large size. Transformers and Accelerate are pinned to
18
- versions compatible with the ZeroGPU runtime dependencies.
 
14
  A coding assistant using Qwen2.5-Coder-3B-Instruct with Hugging Face ZeroGPU.
15
  The model is downloaded from Hugging Face on first startup. Inference requests
16
  use the largest ZeroGPU allocation; xlarge requests consume twice the daily
17
+ GPU quota of the default large size. Transformers is pinned to a version
18
+ compatible with the ZeroGPU runtime dependencies.
app.py CHANGED
@@ -20,7 +20,6 @@ tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
20
  model = AutoModelForCausalLM.from_pretrained(
21
  MODEL_ID,
22
  torch_dtype=torch.float16,
23
- low_cpu_mem_usage=True,
24
  attn_implementation="sdpa",
25
  )
26
  model.to("cuda")
 
20
  model = AutoModelForCausalLM.from_pretrained(
21
  MODEL_ID,
22
  torch_dtype=torch.float16,
 
23
  attn_implementation="sdpa",
24
  )
25
  model.to("cuda")
requirements.txt CHANGED
@@ -1,2 +1 @@
1
  transformers==4.51.3
2
- accelerate==1.7.0
 
1
  transformers==4.51.3