Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -27,7 +27,6 @@ transformer = SD3Transformer2DModel.from_pretrained(
|
|
| 27 |
model_path,
|
| 28 |
subfolder="transformer",
|
| 29 |
torch_dtype=torch.float16,
|
| 30 |
-
use_auth_token=auth_token
|
| 31 |
)
|
| 32 |
transformer = PeftModel.from_pretrained(transformer, "jasperai/flash-sd3")
|
| 33 |
|
|
@@ -40,7 +39,6 @@ if torch.cuda.is_available():
|
|
| 40 |
torch_dtype=torch.float16,
|
| 41 |
text_encoder_3=None,
|
| 42 |
tokenizer_3=None,
|
| 43 |
-
use_auth_token=auth_token
|
| 44 |
)
|
| 45 |
|
| 46 |
pipe = pipe.to(device)
|
|
@@ -51,7 +49,6 @@ else:
|
|
| 51 |
torch_dtype=torch.float16,
|
| 52 |
text_encoder_3=None,
|
| 53 |
tokenizer_3=None,
|
| 54 |
-
use_auth_token=auth_token
|
| 55 |
)
|
| 56 |
pipe = pipe.to(device)
|
| 57 |
|
|
|
|
| 27 |
model_path,
|
| 28 |
subfolder="transformer",
|
| 29 |
torch_dtype=torch.float16,
|
|
|
|
| 30 |
)
|
| 31 |
transformer = PeftModel.from_pretrained(transformer, "jasperai/flash-sd3")
|
| 32 |
|
|
|
|
| 39 |
torch_dtype=torch.float16,
|
| 40 |
text_encoder_3=None,
|
| 41 |
tokenizer_3=None,
|
|
|
|
| 42 |
)
|
| 43 |
|
| 44 |
pipe = pipe.to(device)
|
|
|
|
| 49 |
torch_dtype=torch.float16,
|
| 50 |
text_encoder_3=None,
|
| 51 |
tokenizer_3=None,
|
|
|
|
| 52 |
)
|
| 53 |
pipe = pipe.to(device)
|
| 54 |
|