Update app.py
This commit is contained in:
4
app.py
4
app.py
@@ -69,8 +69,8 @@ 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 1 hour gpu for training
|
# request 10 mins gpu for training
|
||||||
@spaces.GPU(duration=3600)
|
@spaces.GPU(duration=600)
|
||||||
def train_lora(
|
def train_lora(
|
||||||
dataset_path,
|
dataset_path,
|
||||||
project_name,
|
project_name,
|
||||||
|
|||||||
Reference in New Issue
Block a user