Upload app.py
Browse files
app.py
CHANGED
|
@@ -53,6 +53,7 @@ def combine_model_parts(model_dir="split_models", output_file="checkpoints/last.
|
|
| 53 |
print(f"Model combined successfully: {output_file}")
|
| 54 |
return output_file
|
| 55 |
|
|
|
|
| 56 |
def load_model():
|
| 57 |
"""
|
| 58 |
Load the SmollmV2 model and tokenizer.
|
|
|
|
| 53 |
print(f"Model combined successfully: {output_file}")
|
| 54 |
return output_file
|
| 55 |
|
| 56 |
+
@spaces.GPU(enable_queue=True)
|
| 57 |
def load_model():
|
| 58 |
"""
|
| 59 |
Load the SmollmV2 model and tokenizer.
|