Spaces:
Runtime error
Runtime error
Update src/flux/xflux_pipeline.py
Browse files
src/flux/xflux_pipeline.py
CHANGED
@@ -30,7 +30,8 @@ from src.flux.util import (
|
|
30 |
|
31 |
from transformers import CLIPVisionModelWithProjection, CLIPImageProcessor
|
32 |
import spaces
|
33 |
-
|
|
|
34 |
class XFluxPipeline:
|
35 |
def __init__(self, model_type, device, offload: bool = False):
|
36 |
self.device = torch.device(device)
|
|
|
30 |
|
31 |
from transformers import CLIPVisionModelWithProjection, CLIPImageProcessor
|
32 |
import spaces
|
33 |
+
|
34 |
+
|
35 |
class XFluxPipeline:
|
36 |
def __init__(self, model_type, device, offload: bool = False):
|
37 |
self.device = torch.device(device)
|