Spaces:
Running
on
Zero
Running
on
Zero
Update
Browse files
app.py
CHANGED
@@ -52,7 +52,7 @@ def load_t5(device: str | torch.device = "cuda", max_length: int = 512) -> HFEmb
|
|
52 |
def load_clip(device: str | torch.device = "cuda") -> HFEmbedder:
|
53 |
return HFEmbedder("openai/clip-vit-large-patch14", max_length=77, torch_dtype=torch.bfloat16).to("cuda")
|
54 |
|
55 |
-
@spaces.GPU(duration=20)
|
56 |
def load_encoders():
|
57 |
is_schnell = True
|
58 |
t5 = load_t5("cuda", max_length=256 if is_schnell else 512)
|
|
|
52 |
def load_clip(device: str | torch.device = "cuda") -> HFEmbedder:
|
53 |
return HFEmbedder("openai/clip-vit-large-patch14", max_length=77, torch_dtype=torch.bfloat16).to("cuda")
|
54 |
|
55 |
+
# @spaces.GPU(duration=20)
|
56 |
def load_encoders():
|
57 |
is_schnell = True
|
58 |
t5 = load_t5("cuda", max_length=256 if is_schnell else 512)
|