theSure commited on
Commit
b24072b
Β·
verified Β·
1 Parent(s): 7dfbcb9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -54,7 +54,7 @@ image_examples = [
54
 
55
  def load_model(base_model_path, lora_path):
56
  global pipe
57
- transformer = FluxTransformer2DModel.from_pretrained(base_model_path, use_auth_token=os.getenv("flux")), subfolder='transformer', torch_dtype=torch.bfloat16)
58
  gr.Info(str(f"Model loading: {int((40 / 100) * 100)}%"))
59
  # enable image inputs
60
  with torch.no_grad():
 
54
 
55
  def load_model(base_model_path, lora_path):
56
  global pipe
57
+ transformer = FluxTransformer2DModel.from_pretrained(base_model_path, use_auth_token=os.getenv("flux"), subfolder='transformer', torch_dtype=torch.bfloat16)
58
  gr.Info(str(f"Model loading: {int((40 / 100) * 100)}%"))
59
  # enable image inputs
60
  with torch.no_grad():