Spaces:

ajsbsd
/

Qwen2.5-1.5B-Instruct-gkd-demo

Running on Zero

ajsbsd commited on 17 days ago

Commit

33cf6b5

verified ·

1 Parent(s): 5252633

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,6 +2,7 @@ import gradio as gr
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import os
 # --- Configuration ---
 # IMPORTANT: Replace with the path to your locally downloaded model or a Hugging Face model ID.

 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import os
+import spaces
 # --- Configuration ---
 # IMPORTANT: Replace with the path to your locally downloaded model or a Hugging Face model ID.