Hugging Face's logo

Spaces:
weizmannscience
/
tokenflow
Runtime error

App Files Community
4
Linoy Tsaban commited on
Commit
2b854cf
•
1 Parent(s): 0a3b392

Update tokenflow_utils.py

Browse files
Files changed (1) hide show
  1. tokenflow_utils.py +1 -1
tokenflow_utils.py CHANGED
@@ -2,7 +2,7 @@ from typing import Type
2
  import torch
3
  import os
4
 
5
- from util import isinstance_str, batch_cosine_sim
6
 
7
  def register_pivotal(diffusion_model, is_pivotal):
8
  for _, module in diffusion_model.named_modules():
 
2
  import torch
3
  import os
4
 
5
+ from utils import isinstance_str, batch_cosine_sim
6
 
7
  def register_pivotal(diffusion_model, is_pivotal):
8
  for _, module in diffusion_model.named_modules():