Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -35,7 +35,6 @@ client = AsyncInferenceClient()
|
|
35 |
|
36 |
|
37 |
def enable_lora(lora_in, lora_add):
|
38 |
-
pipe.unload_lora_weights()
|
39 |
if not lora_in and not lora_add:
|
40 |
return basemodel
|
41 |
else:
|
@@ -52,8 +51,6 @@ async def generate_image(
|
|
52 |
steps:int=24,
|
53 |
seed:int=-1):
|
54 |
|
55 |
-
pipe.to(device="cuda")
|
56 |
-
|
57 |
if seed == -1:
|
58 |
seed = random.randint(0, MAX_SEED)
|
59 |
seed = int(seed)
|
|
|
35 |
|
36 |
|
37 |
def enable_lora(lora_in, lora_add):
|
|
|
38 |
if not lora_in and not lora_add:
|
39 |
return basemodel
|
40 |
else:
|
|
|
51 |
steps:int=24,
|
52 |
seed:int=-1):
|
53 |
|
|
|
|
|
54 |
if seed == -1:
|
55 |
seed = random.randint(0, MAX_SEED)
|
56 |
seed = int(seed)
|