eeuuia commited on
Commit
502d5a9
·
verified ·
1 Parent(s): c5ceea7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -64,25 +64,25 @@ print("=== Carregando componentes da pipeline... ===")
64
  vae = AutoModel.from_pretrained(
65
  "Lightricks/LTX-Video",
66
  subfolder="vae",
67
- dtype=dtype,
68
  cache_dir=CACHE_DIR
69
  )
70
  text_encoder = AutoModel.from_pretrained(
71
  "Lightricks/LTX-Video",
72
  subfolder="text_encoder",
73
- dtype=dtype,
74
  cache_dir=CACHE_DIR
75
  )
76
  scheduler = AutoModel.from_pretrained(
77
  "Lightricks/LTX-Video",
78
  subfolder="scheduler",
79
- dtype=dtype,
80
  cache_dir=CACHE_DIR
81
  )
82
  tokenizer = AutoModel.from_pretrained(
83
  "Lightricks/LTX-Video",
84
  subfolder="tokenizer",
85
- dtype=dtype,
86
  cache_dir=CACHE_DIR
87
  )
88
 
@@ -94,7 +94,7 @@ if hasattr(scheduler.config, 'use_dynamic_shifting') and scheduler.config.use_dy
94
  transformer = AutoModel.from_pretrained(
95
  "Lightricks/LTX-Video",
96
  subfolder="transformer",
97
- dtype=dtype,
98
  cache_dir=CACHE_DIR
99
  )
100
  transformer.enable_layerwise_casting(
 
64
  vae = AutoModel.from_pretrained(
65
  "Lightricks/LTX-Video",
66
  subfolder="vae",
67
+
68
  cache_dir=CACHE_DIR
69
  )
70
  text_encoder = AutoModel.from_pretrained(
71
  "Lightricks/LTX-Video",
72
  subfolder="text_encoder",
73
+
74
  cache_dir=CACHE_DIR
75
  )
76
  scheduler = AutoModel.from_pretrained(
77
  "Lightricks/LTX-Video",
78
  subfolder="scheduler",
79
+
80
  cache_dir=CACHE_DIR
81
  )
82
  tokenizer = AutoModel.from_pretrained(
83
  "Lightricks/LTX-Video",
84
  subfolder="tokenizer",
85
+
86
  cache_dir=CACHE_DIR
87
  )
88
 
 
94
  transformer = AutoModel.from_pretrained(
95
  "Lightricks/LTX-Video",
96
  subfolder="transformer",
97
+
98
  cache_dir=CACHE_DIR
99
  )
100
  transformer.enable_layerwise_casting(