Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -146,7 +146,6 @@ theme = gr.themes.Monochrome(
|
|
| 146 |
],
|
| 147 |
)
|
| 148 |
|
| 149 |
-
@spaces.GPU(enable_queue=True)
|
| 150 |
def stream(model, code, generate_kwargs):
|
| 151 |
model.to(device)
|
| 152 |
input_ids = tokenizer(code, return_tensors="pt").to("cuda")
|
|
|
|
| 146 |
],
|
| 147 |
)
|
| 148 |
|
|
|
|
| 149 |
def stream(model, code, generate_kwargs):
|
| 150 |
model.to(device)
|
| 151 |
input_ids = tokenizer(code, return_tensors="pt").to("cuda")
|