Spaces:
Running on Zero
Running on Zero
Deploy Qwen2.5 Coder 3B Gradio app
Browse files- README.md +2 -2
- app.py +0 -1
- 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
|
| 18 |
-
|
|
|
|
| 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
|
|
|