Spaces:
				
			
			
	
			
			
		Sleeping
		
	
	
	
			
			
	
	
	
	
		
		
		Sleeping
		
	Commit 
							
							·
						
						928013e
	
1
								Parent(s):
							
							f44876d
								
change the loading args
Browse files- translation.py +1 -1
 
    	
        translation.py
    CHANGED
    
    | 
         @@ -100,7 +100,7 @@ def sentenize_with_fillers(text, splitter, fix_double_space=True, ignore_errors= 
     | 
|
| 100 | 
         | 
| 101 | 
         
             
            class Translator:
         
     | 
| 102 | 
         
             
                def __init__(self):
         
     | 
| 103 | 
         
            -
                    self.model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_URL, low_cpu_mem_usage= 
     | 
| 104 | 
         
             
                    if torch.cuda.is_available():
         
     | 
| 105 | 
         
             
                        self.model.cuda()
         
     | 
| 106 | 
         
             
                    self.tokenizer = NllbTokenizer.from_pretrained(MODEL_URL)
         
     | 
| 
         | 
|
| 100 | 
         | 
| 101 | 
         
             
            class Translator:
         
     | 
| 102 | 
         
             
                def __init__(self):
         
     | 
| 103 | 
         
            +
                    self.model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_URL, low_cpu_mem_usage=True)
         
     | 
| 104 | 
         
             
                    if torch.cuda.is_available():
         
     | 
| 105 | 
         
             
                        self.model.cuda()
         
     | 
| 106 | 
         
             
                    self.tokenizer = NllbTokenizer.from_pretrained(MODEL_URL)
         
     |