Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,7 +1,9 @@
|
|
| 1 |
import gradio as gr
|
| 2 |
import subprocess
|
| 3 |
from huggingface_hub import HfApi
|
|
|
|
| 4 |
|
|
|
|
| 5 |
def merge_and_upload(weight_drop_prob, scaling_factor, base_model, model_to_merge, output_path, repo_name, token):
|
| 6 |
# Construct the command to run hf_merge.py
|
| 7 |
command = [
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
import subprocess
|
| 3 |
from huggingface_hub import HfApi
|
| 4 |
+
import spaces
|
| 5 |
|
| 6 |
+
@spaces.GPU
|
| 7 |
def merge_and_upload(weight_drop_prob, scaling_factor, base_model, model_to_merge, output_path, repo_name, token):
|
| 8 |
# Construct the command to run hf_merge.py
|
| 9 |
command = [
|