Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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")
|
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():
|