stazizov commited on
Commit
6e66743
1 Parent(s): ed315bd

Update src/flux/xflux_pipeline.py

Browse files
Files changed (1) hide show
  1. src/flux/xflux_pipeline.py +2 -1
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)