Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -63,7 +63,7 @@ class HFEmbedder(nn.Module):
|
|
63 |
|
64 |
|
65 |
device = "cuda"
|
66 |
-
t5 = HFEmbedder("
|
67 |
clip = HFEmbedder("openai/clip-vit-large-patch14", max_length=77, torch_dtype=torch.bfloat16).to(device)
|
68 |
ae = AutoencoderKL.from_pretrained("black-forest-labs/FLUX.1-dev", subfolder="vae", torch_dtype=torch.bfloat16).to(device)
|
69 |
# quantize(t5, weights=qfloat8)
|
|
|
63 |
|
64 |
|
65 |
device = "cuda"
|
66 |
+
t5 = HFEmbedder("silveroxides/t5xxl_flan_enc", max_length=512, torch_dtype=torch.bfloat16).to(device)
|
67 |
clip = HFEmbedder("openai/clip-vit-large-patch14", max_length=77, torch_dtype=torch.bfloat16).to(device)
|
68 |
ae = AutoencoderKL.from_pretrained("black-forest-labs/FLUX.1-dev", subfolder="vae", torch_dtype=torch.bfloat16).to(device)
|
69 |
# quantize(t5, weights=qfloat8)
|