Update app.py
This commit is contained in:
2
app.py
2
app.py
@@ -70,7 +70,7 @@ def upload_and_prepare_dataset(files, dataset_name, trigger_word):
|
|||||||
return f"✅ ready: {image_count} images in {dataset_name}", str(dataset_path), dataset_name
|
return f"✅ ready: {image_count} images in {dataset_name}", str(dataset_path), dataset_name
|
||||||
|
|
||||||
# request 10 mins gpu for training
|
# request 10 mins gpu for training
|
||||||
@spaces.GPU(duration=600)
|
@spaces.GPU(duration=200)
|
||||||
def train_lora(
|
def train_lora(
|
||||||
dataset_path,
|
dataset_path,
|
||||||
project_name,
|
project_name,
|
||||||
|
|||||||
Reference in New Issue
Block a user