vilarin commited on
Commit
ee571c8
1 Parent(s): 8f3c0da

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -3
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)