Spaces:
vilarin
/
Running on Zero

vilarin commited on
Commit
0756ffd
1 Parent(s): 4eadc27

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -6
app.py CHANGED
@@ -32,12 +32,9 @@ JS = """function () {
32
  }
33
  }"""
34
 
35
- if torch.cuda.is_available():
36
- pipe = FluxPipeline.from_pretrained(model, torch_dtype=torch.bfloat16)
37
- pipe.load_lora_weights(default_lora, weight_name = default_weight_name) # default load lora
38
- pipe.fuse_lora(lora_scale=0.9)
39
-
40
-
41
 
42
  def scrape_lora_link(url):
43
  try:
 
32
  }
33
  }"""
34
 
35
+ pipe = FluxPipeline.from_pretrained(model, torch_dtype=torch.bfloat16)
36
+ pipe.load_lora_weights(default_lora, weight_name = default_weight_name) # default load lora
37
+ pipe.fuse_lora(lora_scale=0.9)
 
 
 
38
 
39
  def scrape_lora_link(url):
40
  try: